Searched refs:rhs_instruction (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 731 HloInstruction* rhs_instruction = in TEST_P() local
739 rhs_instruction, dot_dnums, in TEST_P()
777 HloInstruction* rhs_instruction = in TEST_P() local
785 rhs_instruction, dot_dnums, in TEST_P()
821 HloInstruction* rhs_instruction = in TEST_P() local
829 rhs_instruction, dot_dnums, in TEST_P()
858 HloInstruction* rhs_instruction = in TEST_P() local
871 rhs_instruction, dot_dnums, in TEST_P()
901 HloInstruction* rhs_instruction = in TEST_P() local
928 shape, lhs_instruction, rhs_instruction, /*feature_group_count=*/1, in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 469 auto rhs_instruction = dot->operand(1); in HandleDot() local
472 const llvm_ir::IrArray& rhs_array = GetIrArray(*rhs_instruction, *dot); in HandleDot()
475 const Shape& rhs_shape = rhs_instruction->shape(); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 539 HloInstruction* rhs_instruction = in ConvArray4DGeneralDimensionsDilated() local
546 shape, lhs_instruction, rhs_instruction, /*feature_group_count=*/1, in ConvArray4DGeneralDimensionsDilated()
|