Searched defs:new_source_dim (Results 1 – 1 of 1) 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