Home
last modified time | relevance | path

Searched refs:GatherParallelOutputDims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h211 absl::InlinedVector<int64, 1> GatherParallelOutputDims(
Dhlo_sharding_util.cc900 auto output_parallel_dims = GatherParallelOutputDims(gather, parallel_dims); in GatherParallelDataOperandSharding()
1337 absl::InlinedVector<int64, 1> GatherParallelOutputDims( in GatherParallelOutputDims() function
Dsharding_propagation.cc455 hlo_sharding_util::GatherParallelOutputDims(*instruction, parallel_dims); in InferGatherParallelShardingFromOperands()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc338 hlo_sharding_util::GatherParallelOutputDims(*gather, *parallel_dims); in PartitionIndexParallelDimensions()