Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddefuser.cc48 HloComputation* fused_computation = in Defuse() local
57 defused_instructions[fused_computation->parameter_instruction(i)] = in Defuse()
66 fused_computation->MakeInstructionPostOrder()) { in Defuse()
87 return module->RemoveEmbeddedComputation(fused_computation); in Defuse()
Dhlo_module_test.cc117 HloComputation* fused_computation; in TEST_F() local
123 fused_computation = module->AddEmbeddedComputation(b.Build()); in TEST_F()
133 /*operands=*/{input}, fused_computation)); in TEST_F()
Dhlo_cse_test.cc507 auto fused_computation = in TEST_F() local
513 EXPECT_EQ(5, fused_computation->instruction_count()); in TEST_F()
516 EXPECT_EQ(4, fused_computation->instruction_count()); in TEST_F()
518 auto root = fused_computation->root_instruction(); in TEST_F()
Dhlo_verifier.cc1181 HloComputation* fused_computation = fusion->fused_instructions_computation(); in CheckFusionInstruction() local
1182 if (fusion != fused_computation->FusionInstruction()) { in CheckFusionInstruction()
1197 for (auto* instruction : fused_computation->instructions()) { in CheckFusionInstruction()
1242 if (fused_computation != user->parent()) { in CheckFusionInstruction()
Dhlo_instruction_test.cc778 auto* fused_computation = fusion->fused_instructions_computation(); in TEST_F() local
779 EXPECT_THAT(fusion->called_computations(), ElementsAre(fused_computation)); in TEST_F()
782 EXPECT_THAT(fusion->called_computations(), ElementsAre(fused_computation)); in TEST_F()
785 EXPECT_THAT(fusion->called_computations(), ElementsAre(fused_computation)); in TEST_F()
Dhlo_parser_test.cc791 %fused_computation (constant.param_0: f32[3,2,1,1], constant.1.param_1: f32[2]) -> f32[3,2,1,1] { in CreateTestCases()
801 …usion(f32[3,2,1,1]{3,2,1,0} %constant, f32[2]{0} %constant.1), kind=kLoop, calls=%fused_computation in CreateTestCases()
Dhlo_instruction.cc331 auto* fused_computation = in CreateFromProto() local
333 TF_RET_CHECK(fused_computation != nullptr) in CreateFromProto()
336 CreateFusion(shape, fusion_kind, all_operands(), fused_computation); in CreateFromProto()