Searched defs:new_source_dim (Results 1 – 2 of 2) sorted by relevance
519 int64_t new_source_dim = in ReshapeToRemoveDegenerateDims() local574 int64_t new_source_dim = [&]() { in ReshapeToAddDegenerateDims() local1061 int64_t new_source_dim = dim_numbers.lhs_batch_dimensions_size() + in ComputeArrayForDotWithIndexedLhs() local1097 int64_t new_source_dim = dim_numbers.lhs_batch_dimensions_size() + in ComputeArrayForDotWithIndexedRhs() local
1591 int64_t new_source_dim = in ReshardWithAllToAll() local