Home
last modified time | relevance | path

Searched refs:dot_shape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc272 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in RunDotOutputFusion() local
277 HloInstruction::CreateParameter(1, dot_shape, "param1")); in RunDotOutputFusion()
281 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs)); in RunDotOutputFusion()
285 dot_shape, HloOpcode::kAdd, dot_result, addend)); in RunDotOutputFusion()
288 dot_shape, HloOpcode::kAdd, addend, dot_result)); in RunDotOutputFusion()
295 dot_shape, HloInstruction::FusionKind::kOutput, add_result)); in RunDotOutputFusion()
310 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape)); in RunDotOutputFusion()
312 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape)); in RunDotOutputFusion()
Dcpu_instruction_fusion_test.cc638 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in CreateComputationForDotAddOutputFusionTest() local
645 HloInstruction::CreateParameter(2, dot_shape, "param2")); in CreateComputationForDotAddOutputFusionTest()
648 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs)); in CreateComputationForDotAddOutputFusionTest()
650 HloInstruction::CreateBinary(dot_shape, HloOpcode::kAdd, dot, addend)); in CreateComputationForDotAddOutputFusionTest()
655 HloInstruction::CreateOutfeed(dot_shape, dot, token, "no_config")); in CreateComputationForDotAddOutputFusionTest()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc41 const Shape& dot_shape = dot->shape(); in DecomposeBatchDot() local
88 PrimitiveType dot_type = dot_shape.element_type(); in DecomposeBatchDot()
89 const int64 dot_rows = dot_shape.dimensions(num_batch_dims + 0); in DecomposeBatchDot()
90 const int64 dot_cols = dot_shape.dimensions(num_batch_dims + 1); in DecomposeBatchDot()
155 HloInstruction::CreateReshape(dot_shape, concat)); in DecomposeBatchDot()
Dhlo_creation_utils.cc223 Shape dot_shape, in MakeDotHlo()
226 dot_shape, lhs, rhs, dim_numbers, precision_config)); in MakeDotHlo()
Dalgebraic_simplifier_test.cc4335 Shape dot_shape = ShapeUtil::MakeShape(element_type, output_dims); in TEST_P() local
4356 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4391 Shape dot_shape = ShapeUtil::MakeShape(element_type, {m, n}); in TEST_P() local
4414 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4488 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
4490 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4498 ShapeUtil::Equal(computation->root_instruction()->shape(), dot_shape)); in TEST_P()
4553 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
4555 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
4562 ShapeUtil::Equal(computation->root_instruction()->shape(), dot_shape)); in TEST_P()
[all …]
Dhlo_cost_analysis.cc241 const Shape& dot_shape = dot->shape(); in HandleDot() local
251 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width; in HandleDot()
Dhlo_evaluator.cc373 Shape dot_shape, in EvaluateDotOp()
377 HloInstruction::CreateDot(dot_shape, lhs_instr.get(), rhs_instr.get(), in EvaluateDotOp()