Fix max number of dimensions used for std::bitset.

Internally, we add up to 2 dimensions on top of the limit in the public API.

PiperOrigin-RevId: 832141894

Author: dsharletg

ynnpack/subgraph/subgraph.h

@@ -28,7 +28,10 @@
 
 namespace ynn {
 
-using axes_set = std::bitset<YNN_MAX_TENSOR_RANK>;
+// dot packing splits + transposes 2 dimensions.
+constexpr size_t ynn_internal_extra_dims = 2;
+
+using axes_set = std::bitset<YNN_MAX_TENSOR_RANK + ynn_internal_extra_dims>;
 
 // Define a transpose node, optionally using a slinky copy that may alias even
 // if dimension 0 is not stride 1 in the result.