Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h153 const HloSharding& data_operand_sharding, const HloInstruction& hlo,
Dhlo_sharding_util.cc947 const HloSharding& data_operand_sharding, const HloInstruction& hlo, in GatherOutputShardingFromDataOperand() argument
957 operand_shape, data_operand_sharding, output_shape, collapsed_slice_dims, in GatherOutputShardingFromDataOperand()