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.cc1406 absl::InlinedVector<int64, 1> indices_parallel_dims; in GetGatherBatchParallelDims() local
1440 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.cc1882 auto& indices_parallel_dims = parallel_dims.indices_parallel_dims; in GatherOperandsShardedAcrossParallelDims() local