Home
last modified time | relevance | path

Searched refs:data_operand_sharding (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.h159 const HloSharding& data_operand_sharding, const HloInstruction& hlo,
H A Dhlo_sharding_util.cc1062 const HloSharding& data_operand_sharding, const HloInstruction& hlo, in GatherOutputShardingFromDataOperand() argument
1073 operand_shape, data_operand_sharding, output_shape, collapsed_slice_dims, in GatherOutputShardingFromDataOperand()