Home
last modified time | relevance | path

Searched refs:GatherOutputAlignedOperandParallelDims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h217 absl::InlinedVector<int64, 1> GatherOutputAlignedOperandParallelDims(
Dhlo_sharding_util.cc902 GatherOutputAlignedOperandParallelDims(gather, parallel_dims); in GatherParallelDataOperandSharding()
1357 absl::InlinedVector<int64, 1> GatherOutputAlignedOperandParallelDims( in GatherOutputAlignedOperandParallelDims() function
Dsharding_propagation.cc461 hlo_sharding_util::GatherOutputAlignedOperandParallelDims( in InferGatherParallelShardingFromOperands()