Searched defs:output_sharding_dim (Results 1 – 1 of 1) sorted by relevance
389 int64 output_sharding_dim, int64 rhs_shape_size, int64 lhs_shape_size, in GetWindowedEinsumConfiguration()585 const int64 output_sharding_dim = in PartitionBaseCase() local2395 const int64 output_sharding_dim = FirstShardingDimWithPartitionOfSize( in LhsIsBestMatchForNonContractingPartitioning() local