Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.h245 absl::InlinedVector<int64_t, 1> GatherOutputAlignedOperandParallelDims(
H A Dhlo_sharding_util.cc1019 GatherOutputAlignedOperandParallelDims(gather, parallel_dims); in GatherParallelDataOperandSharding()
1580 absl::InlinedVector<int64_t, 1> GatherOutputAlignedOperandParallelDims( in GatherOutputAlignedOperandParallelDims() function
H A Dsharding_propagation.cc516 hlo_sharding_util::GatherOutputAlignedOperandParallelDims( in InferGatherParallelShardingFromOperands()