Searched refs:indices_map (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 491 DotDimensionIndexMapping indices_map = ComputeDimensionIndexMapping( in PartitionBaseCase() local
496 lhs_sharding, indices_map.lhs_to_rhs_indices, in PartitionBaseCase()
497 indices_map.rhs_to_lhs_indices); in PartitionBaseCase()
500 rhs_sharding, indices_map.rhs_to_lhs_indices, in PartitionBaseCase()
501 indices_map.lhs_to_rhs_indices); in PartitionBaseCase()
504 lhs_sharding, indices_map.lhs_to_output_indices, in PartitionBaseCase()
505 indices_map.output_to_lhs_indices); in PartitionBaseCase()
508 rhs_sharding, indices_map.rhs_to_output_indices, in PartitionBaseCase()
509 indices_map.output_to_rhs_indices); in PartitionBaseCase()
512 output_sharding, indices_map.output_to_lhs_indices, in PartitionBaseCase()
[all …]
|