I ran into the same issue.
The documentation (https://tomcat.apache.org/tomcat-11.0-doc/config/context.html) says:
usePartitionedShould
the Partitioned flag be set on session cookies? Defaults to false.
It should work by adding the attribute usePartitioned="true" to the <Context> element in your context.xml. Unfortunately, it doesn't work. I submitted the following bug: https://bz.apache.org/bugzilla/show_bug.cgi?id=69836