Searched refs:data_operand (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | value_inference_test.cc | 292 auto data_operand = ConstantR1<int32_t>(&b, {1, 2}); in TEST_F() local
298 auto gather = Gather(data_operand, indices, dim_numbers, {1}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 1451 HloInstruction::CreateAddDependency(HloInstruction* data_operand, in CreateAddDependency() argument
1454 new HloInstruction(HloOpcode::kAddDependency, data_operand->shape())); in CreateAddDependency()
1455 instruction->AppendOperand(data_operand); in CreateAddDependency()
1758 const Shape& shape, HloInstruction* data_operand, in CreateDynamicReshape() argument
1761 ShapeUtil::ElementsIn(data_operand[0].shape())) in CreateDynamicReshape()
1763 << " operand: " << ShapeUtil::HumanString(data_operand[0].shape()); in CreateDynamicReshape()
1765 return std::make_unique<HloDynamicReshapeInstruction>(shape, data_operand, in CreateDynamicReshape()
|
| D | hlo_instruction.h | 1020 const Shape& shape, HloInstruction* data_operand,
1173 HloInstruction* data_operand, HloInstruction* token_operand);
|
| D | hlo_instructions.cc | 1271 const Shape& shape, HloInstruction* data_operand, in HloDynamicReshapeInstruction() argument
1274 AppendOperand(data_operand); in HloDynamicReshapeInstruction()
|
| D | hlo_instructions.h | 967 const Shape& shape, HloInstruction* data_operand,
|