Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.h35 absl::InlinedVector<int64_t, 1> indices_parallel_dims; member
H A Dhlo_sharding_util.cc1529 absl::InlinedVector<int64_t, 1> indices_parallel_dims; in GetGatherBatchParallelDims() local
1563 auto indices_parallel_dims = parallel_dim.indices_parallel_dims; in GatherParallelOutputDims() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dgather_scatter_handler.cc499 auto indices_parallel_dims = parallel_dims->indices_parallel_dims; in PartitionGatherIndexParallelDimensions() local
H A Dspmd_partitioner_util.cc1709 auto& indices_parallel_dims = parallel_dims.indices_parallel_dims; in GatherOperandsShardedAcrossParallelDims() local