Searched refs:data_operand (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dynamism_inference_test.cc | 299 auto data_operand = ConstantR1<int32>(&b, {1, 2}); in TEST_F() local
305 auto gather = Gather(data_operand, indices, dim_numbers, {1}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 1177 HloInstruction::CreateAddDependency(HloInstruction* data_operand, in CreateAddDependency() argument
1180 new HloInstruction(HloOpcode::kAddDependency, data_operand->shape())); in CreateAddDependency()
1181 instruction->AppendOperand(data_operand); in CreateAddDependency()
1459 const Shape& shape, HloInstruction* data_operand, in CreateDynamicReshape() argument
1462 ShapeUtil::ElementsIn(data_operand[0].shape())) in CreateDynamicReshape()
1464 << " operand: " << ShapeUtil::HumanString(data_operand[0].shape()); in CreateDynamicReshape()
1466 return absl::make_unique<HloDynamicReshapeInstruction>(shape, data_operand, in CreateDynamicReshape()
|
| D | hlo_instruction.h | 905 const Shape& shape, HloInstruction* data_operand,
1042 HloInstruction* data_operand, HloInstruction* token_operand);
|
| D | hlo_instructions.cc | 1082 const Shape& shape, HloInstruction* data_operand, in HloDynamicReshapeInstruction() argument
1085 AppendOperand(data_operand); in HloDynamicReshapeInstruction()
|
| D | hlo_instructions.h | 720 const Shape& shape, HloInstruction* data_operand,
|