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
275 dot_shape = ShapeUtil::MakeShape(F32, {n}); in RunDotOutputFusion()
278 dot_shape = ShapeUtil::MakeShape(F32, {m}); in RunDotOutputFusion()
284 HloInstruction::CreateParameter(1, dot_shape, "param1")); in RunDotOutputFusion()
288 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs)); in RunDotOutputFusion()
292 dot_shape, HloOpcode::kAdd, dot_result, addend)); in RunDotOutputFusion()
295 dot_shape, HloOpcode::kAdd, addend, dot_result)); in RunDotOutputFusion()
302 dot_shape, HloInstruction::FusionKind::kOutput, add_result)); in RunDotOutputFusion()
317 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape)); in RunDotOutputFusion()
319 ShapeLayout(LayoutUtil::GetWithDefaultLayout(dot_shape)); in RunDotOutputFusion()
Dcpu_instruction_fusion_test.cc640 Shape dot_shape = ShapeUtil::MakeShape(F32, {m, n}); in CreateComputationForDotAddOutputFusionTest() local
643 dot_shape = ShapeUtil::MakeShape(F32, {n}); in CreateComputationForDotAddOutputFusionTest()
646 dot_shape = ShapeUtil::MakeShape(F32, {m}); in CreateComputationForDotAddOutputFusionTest()
654 HloInstruction::CreateParameter(2, dot_shape, "param2")); in CreateComputationForDotAddOutputFusionTest()
657 builder.AddInstruction(CreateCanonicalDot(dot_shape, dot_lhs, dot_rhs)); in CreateComputationForDotAddOutputFusionTest()
659 HloInstruction::CreateBinary(dot_shape, HloOpcode::kAdd, dot, addend)); in CreateComputationForDotAddOutputFusionTest()
664 HloInstruction::CreateOutfeed(dot_shape, dot, token, "no_config")); in CreateComputationForDotAddOutputFusionTest()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc294 Shape dot_shape, in MakeDotHlo()
298 dot_shape, lhs, rhs, dim_numbers, precision_config)); in MakeDotHlo()
Dalgebraic_simplifier_test.cc5192 Shape dot_shape = ShapeUtil::MakeShape(element_type, output_dims); in TEST_P() local
5213 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5248 Shape dot_shape = ShapeUtil::MakeShape(element_type, {m, n}); in TEST_P() local
5271 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5352 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
5354 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5362 ShapeUtil::Equal(computation->root_instruction()->shape(), dot_shape)); in TEST_P()
5417 Shape dot_shape = ShapeUtil::MakeShape(F32, {spec.m, spec.n}); in TEST_P() local
5419 dot_shape, lhs, rhs, dot_dnums, DefaultPrecisionConfig(2))); in TEST_P()
5426 ShapeUtil::Equal(computation->root_instruction()->shape(), dot_shape)); in TEST_P()
[all …]
Dhlo_cost_analysis.cc325 const Shape& dot_shape = dot->shape(); in HandleDot() local
335 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width; in HandleDot()
Dhlo_evaluator.cc391 TF_ASSIGN_OR_RETURN(Shape dot_shape, in EvaluateDotOp()
397 HloInstruction::CreateDot(dot_shape, lhs_instr.get(), rhs_instr.get(), in EvaluateDotOp()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc527 TF_ASSIGN_OR_RETURN(Shape dot_shape, builder->GetShape(dot)); in Einsum()
532 new_dims.push_back(dot_shape.dimensions(dot_dim)); in Einsum()