Searched refs:new_source_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 513 int64 new_source_dim = in ReshapeToRemoveDegenerateDims() local
517 new_source, new_indices, new_source_dim, in ReshapeToRemoveDegenerateDims()
568 int64 new_source_dim = [&]() { in ReshapeToAddDegenerateDims() local
574 if (new_source_shape_dims[new_source_dim] != 1) { in ReshapeToAddDegenerateDims()
583 InsertAt(&new_source_shape_dims, /*index=*/new_source_dim, in ReshapeToAddDegenerateDims()
595 new_source, operand->indices(), new_source_dim, in ReshapeToAddDegenerateDims()
1057 int64 new_source_dim = dim_numbers.lhs_batch_dimensions_size() + in ComputeArrayForDotWithIndexedLhs() local
1062 new_source, lhs->indices(), new_source_dim, in ComputeArrayForDotWithIndexedLhs()
1093 int64 new_source_dim = dim_numbers.lhs_batch_dimensions_size() + in ComputeArrayForDotWithIndexedRhs() local
1098 new_source, rhs->indices(), new_source_dim, in ComputeArrayForDotWithIndexedRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1190 int64 new_source_dim = in ReshardWithAllToAll() local
1197 if (i == new_source_dim) { in ReshardWithAllToAll()
|