Searched defs:output_slice_dims (Results 1 – 1 of 1) sorted by relevance
2176 std::vector<int64> output_slice_dims; in GetDotGroupPartitionContractingOutputShardings() local2341 std::vector<int64> output_slice_dims; in PartitionDotGroupOnContracting() local2692 std::vector<int64> output_slice_dims; in PrioritizeContractingDimensionsPartitioning() local