Home
last modified time | relevance | path

Searched defs:indices_parallel_dims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h34 absl::InlinedVector<int64, 1> indices_parallel_dims; member
Dhlo_sharding_util.cc1306 absl::InlinedVector<int64, 1> indices_parallel_dims; in GetGatherBatchParallelDims() local
1340 auto indices_parallel_dims = parallel_dim.indices_parallel_dims; in GatherParallelOutputDims() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc335 auto indices_parallel_dims = parallel_dims->indices_parallel_dims; in PartitionIndexParallelDimensions() local
Dspmd_partitioner_util.cc1835 auto& indices_parallel_dims = parallel_dims.indices_parallel_dims; in GatherOperandsShardedAcrossParallelDims() local