Searched defs:new_source_dim (Results 1 – 2 of 2) sorted by relevance
513 int64 new_source_dim = in ReshapeToRemoveDegenerateDims() local568 int64 new_source_dim = [&]() { in ReshapeToAddDegenerateDims() local1057 int64 new_source_dim = dim_numbers.lhs_batch_dimensions_size() + in ComputeArrayForDotWithIndexedLhs() local1093 int64 new_source_dim = dim_numbers.lhs_batch_dimensions_size() + in ComputeArrayForDotWithIndexedRhs() local
1190 int64 new_source_dim = in ReshardWithAllToAll() local