Home
last modified time | relevance | path

Searched defs:output_parallel_dims (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dsharding_propagation.cc458 absl::Span<const int64_t> output_parallel_dims) { in InferGatherParallelShardingFromOperands()
509 auto output_parallel_dims = in InferGatherParallelShardingFromOperands() local
H A Dhlo_sharding_util.cc1017 auto output_parallel_dims = GatherParallelOutputDims(gather, parallel_dims); in GatherParallelDataOperandSharding() local
1562 absl::InlinedVector<int64_t, 1> output_parallel_dims; in GatherParallelOutputDims() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dgather_scatter_handler.cc501 auto output_parallel_dims = in PartitionGatherIndexParallelDimensions() local