| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation_test.cc | 85 std::unique_ptr<HloInstruction> CreateDot(const Shape& shape, in CreateDot() function in xla::BFloat16PropagationTest
91 return HloInstruction::CreateDot(shape, lhs, rhs, dot_dnums, in CreateDot()
120 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), xpose, a)); in TEST_F()
181 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), xpose, rw)); in TEST_F()
219 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, gte0, gte1)); in TEST_F()
243 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, a, b)); in TEST_F()
296 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), lhs, rhs)); in TEST_F()
338 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), lhs, rhs)); in TEST_F()
364 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, add, add)); in TEST_F()
408 HloInstruction* dot = builder_f1.AddInstruction(CreateDot(shape, a_f1, b_f1)); in TEST_F()
[all …]
|
| D | dot_decomposer.cc | 140 HloInstruction::CreateDot(dot_shape_r2, lhs_slice_r2, rhs_slice_r2, in DecomposeBatchDot()
277 HloInstruction* dot = computation->AddInstruction(HloInstruction::CreateDot( in CanonicalizeDot()
|
| D | heap_simulator_test.cc | 494 auto dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
530 auto dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
568 auto dot0 = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
570 auto dot1 = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
609 auto dot0 = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
611 auto dot1 = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
|
| D | hlo_instruction_test.cc | 1152 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
1193 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
1244 auto dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
1325 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
1490 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
1532 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
1588 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
|
| D | hlo_computation_test.cc | 557 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
592 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
628 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
|
| D | transpose_folding.cc | 110 std::unique_ptr<HloInstruction> new_dot = HloInstruction::CreateDot( in FoldTransposeIntoDot()
|
| D | bfloat16_normalization_test.cc | 357 HloInstruction::CreateDot(bf16_shape, a, b, dot_dnums, precision_config)); in TEST_F()
|
| D | hlo_creation_utils.cc | 225 return computation->AddInstruction(HloInstruction::CreateDot( in MakeDotHlo()
|
| D | hlo_evaluator_test.cc | 738 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
784 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
828 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
870 b.AddInstruction(HloInstruction::CreateDot(shape, lhs_instruction, in TEST_P()
|
| D | dynamic_dimension_inference_test.cc | 262 HloInstruction::CreateDot(xz_shape, a_param, b_param, dot_dnums, in TEST_F()
|
| D | algebraic_simplifier_test.cc | 3770 builder.AddInstruction(HloInstruction::CreateDot(r1f32, x, y, dot_dnums, in TEST_F()
4355 builder.AddInstruction(HloInstruction::CreateDot( in TEST_P()
4413 builder.AddInstruction(HloInstruction::CreateDot( in TEST_P()
4489 builder.AddInstruction(HloInstruction::CreateDot( in TEST_P()
4554 builder.AddInstruction(HloInstruction::CreateDot( in TEST_P()
4677 builder.AddInstruction(HloInstruction::CreateDot( in TEST_P()
4750 builder.AddInstruction(HloInstruction::CreateDot( in TEST_P()
|
| D | algebraic_simplifier.cc | 1488 HloInstruction::CreateDot(dot.shape(), new_dot_lhs, new_dot_rhs, in OptimizeDotOfConcatHelper()
1595 HloInstruction::CreateDot(memoized_shape, left_operand, right_operand, in OptimizeDotOfGather()
1810 auto new_dot = computation_->AddInstruction(HloInstruction::CreateDot( in HandleDot()
3632 auto dot = computation_->AddInstruction(HloInstruction::CreateDot( in SimplifyConvToDot()
|
| D | hlo_instruction.h | 462 static std::unique_ptr<HloInstruction> CreateDot(
|
| D | buffer_assignment_test.cc | 1598 auto dot_ab = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
1600 auto dot_bc = builder.AddInstruction(HloInstruction::CreateDot( in TEST_F()
|
| D | tuple_points_to_analysis_test.cc | 1133 HloInstruction::CreateDot(data_shape, a, b, dot_dnums, precision_config)); in TEST_F()
|
| D | hlo_evaluator.cc | 377 HloInstruction::CreateDot(dot_shape, lhs_instr.get(), rhs_instr.get(), in EvaluateDotOp()
|
| D | hlo_dataflow_analysis_test.cc | 2512 HloInstruction::CreateDot(data_shape, a, b, dot_dnums, precision_config)); in TEST_F()
|
| D | hlo_instruction.cc | 815 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateDot( in CreateDot() function in xla::HloInstruction
|
| D | hlo_parser.cc | 1610 instruction = builder->AddInstruction(HloInstruction::CreateDot( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multioutput_fusion_test.cc | 92 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in RunTest2D()
154 HloInstruction* dot = builder.AddInstruction(HloInstruction::CreateDot( in RunTest1D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 45 return HloInstruction::CreateDot(shape, lhs, rhs, dot_dnums, in MakeDot()
|