Searched refs:lhs_instruction (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 398 auto lhs_instruction = b.AddInstruction(HloInstruction::CreateBinary( in TEST_F() local
404 lhs_instruction, param_rhs2)); in TEST_F()
725 HloInstruction* lhs_instruction = in TEST_P() local
738 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
765 HloInstruction* lhs_instruction = in TEST_P() local
784 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
809 HloInstruction* lhs_instruction = in TEST_P() local
828 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
852 HloInstruction* lhs_instruction = in TEST_P() local
870 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 468 auto lhs_instruction = dot->operand(0); in HandleDot() local
471 const llvm_ir::IrArray& lhs_array = GetIrArray(*lhs_instruction, *dot); in HandleDot()
474 const Shape& lhs_shape = lhs_instruction->shape(); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 537 HloInstruction* lhs_instruction = in ConvArray4DGeneralDimensionsDilated() local
546 shape, lhs_instruction, rhs_instruction, /*feature_group_count=*/1, in ConvArray4DGeneralDimensionsDilated()
|