Searched refs:rhs_instruction (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.cc | 106 const HloInstruction* rhs_instruction = convolution->operand(1); in AddBackendConstraints() local
115 Shape filter_shape(RowMajorShape(rhs_instruction->shape())); in AddBackendConstraints()
145 const HloInstruction* rhs_instruction = dot->operand(1); in AddBackendConstraints() local
146 Shape rhs_shape(RowMajorShape(rhs_instruction->shape())); in AddBackendConstraints()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 630 HloInstruction* rhs_instruction = in TEST_P() local
638 rhs_instruction, dot_dnums)); in TEST_P()
675 HloInstruction* rhs_instruction = in TEST_P() local
683 rhs_instruction, dot_dnums)); in TEST_P()
718 HloInstruction* rhs_instruction = in TEST_P() local
726 rhs_instruction, dot_dnums)); in TEST_P()
752 HloInstruction* rhs_instruction = in TEST_P() local
779 shape, lhs_instruction, rhs_instruction, window, dnums)); in TEST_P()
814 HloInstruction* rhs_instruction = in TEST_P() local
833 shape, lhs_instruction, rhs_instruction, window, dnums)); in TEST_P()
[all …]
|
| D | hlo_cost_analysis.cc | 382 auto rhs_instruction = convolution->operand(1); in HandleConvolution() local
391 ? ShapeUtil::ElementsIn(rhs_instruction->shape()) / output_features in HandleConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 443 auto rhs_instruction = dot->operand(1); in HandleDot() local
446 const llvm_ir::IrArray& rhs_array = GetIrArray(*rhs_instruction, *dot); in HandleDot()
449 const Shape& rhs_shape = rhs_instruction->shape(); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 570 HloInstruction* rhs_instruction = in ConvArray4DGeneralDimensionsDilated() local
574 shape, lhs_instruction, rhs_instruction, window, dnums)); in ConvArray4DGeneralDimensionsDilated()
|