Searched refs:fusion_instruction (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | defuser.cc | 45 Status Defuse(HloInstruction* fusion_instruction) { in Defuse() argument
46 VLOG(2) << "Defusing instruction: " << fusion_instruction->ToString(); in Defuse()
49 fusion_instruction->fused_instructions_computation(); in Defuse()
56 for (int64 i = 0; i < fusion_instruction->operand_count(); ++i) { in Defuse()
58 fusion_instruction->mutable_operand(i); in Defuse()
75 fusion_instruction->parent()->AddInstruction( in Defuse()
81 TF_RETURN_IF_ERROR(fusion_instruction->ReplaceAllUsesWith( in Defuse()
82 defused_instructions.at(fusion_instruction->fused_expression_root()))); in Defuse()
84 HloModule* module = fusion_instruction->parent()->parent(); in Defuse()
86 fusion_instruction->parent()->RemoveInstruction(fusion_instruction)); in Defuse()
[all …]
|
| D | instruction_fusion.cc | 494 HloInstruction* fusion_instruction; in Run() local
500 fusion_instruction = Fuse(operand, instruction); in Run()
504 fusion_instruction = FuseIntoMultiOutput(operand, instruction); in Run()
509 fusion_queue->OnFusingInstruction(fusion_instruction, operand, in Run()
521 if (fusion_instruction != instruction) { in Run()
537 HloInstruction* fusion_instruction; in AddFusionInstruction() local
540 fusion_instruction = consumer; in AddFusionInstruction()
541 if (kind != fusion_instruction->fusion_kind()) { in AddFusionInstruction()
542 fusion_instruction->set_fusion_kind(kind); in AddFusionInstruction()
545 fusion_instruction = computation_->AddInstruction( in AddFusionInstruction()
[all …]
|
| D | hlo_computation.h | 72 HloInstruction* fusion_instruction = nullptr)
75 fusion_instruction_(fusion_instruction) {} in name_()
399 void SetFusionInstruction(HloInstruction* fusion_instruction) { in SetFusionInstruction() argument
400 fusion_instruction_ = fusion_instruction; in SetFusionInstruction()
424 HloInstruction* root_instruction, HloInstruction* fusion_instruction);
435 HloInstruction* fusion_instruction);
|
| D | hlo_computation.cc | 70 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation() argument
74 fusion_instruction_(fusion_instruction) { in HloComputation()
597 HloInstruction* fusion_instruction) { in FuseInstructionsInto() argument
598 CHECK_EQ(HloOpcode::kFusion, fusion_instruction->opcode()); in FuseInstructionsInto()
600 TF_CHECK_OK(root->ReplaceAllUsesWith(fusion_instruction)); in FuseInstructionsInto()
602 set_root_instruction(fusion_instruction); in FuseInstructionsInto()
607 fusion_instruction->FuseInstruction(instruction); in FuseInstructionsInto()
618 HloInstruction* fusion_instruction = AddInstruction( in CreateFusionInstruction() local
620 FuseInstructionsInto(instructions_to_fuse, fusion_instruction); in CreateFusionInstruction()
621 return fusion_instruction; in CreateFusionInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 76 auto fusion_instruction = computation->root_instruction(); in TEST_F() local
77 EXPECT_EQ(HloOpcode::kFusion, fusion_instruction->opcode()); in TEST_F()
79 fusion_instruction->fused_expression_root()->opcode()); in TEST_F()
82 EXPECT_EQ(4, fusion_instruction->fused_instruction_count()); in TEST_F()
123 auto fusion_instruction = computation->root_instruction(); in TEST_F() local
124 EXPECT_EQ(HloOpcode::kFusion, fusion_instruction->opcode()); in TEST_F()
126 fusion_instruction->fused_expression_root()->opcode()); in TEST_F()
129 EXPECT_EQ(8, fusion_instruction->fused_instruction_count()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 293 HloInstruction* fusion_instruction = in RunDotOutputFusion() local
297 computation->ReplaceInstruction(add_result, fusion_instruction)); in RunDotOutputFusion()
300 fusion_instruction->fused_instructions_computation()->root_instruction(); in RunDotOutputFusion()
301 HloInstruction* fused_dot = fusion_instruction->FuseInstruction(dot_result); in RunDotOutputFusion()
315 fusion_instruction->operand(fused_dot->operand(0)->parameter_number()); in RunDotOutputFusion()
317 fusion_instruction->operand(fused_dot->operand(1)->parameter_number()); in RunDotOutputFusion()
318 result.addend_fusion_param = fusion_instruction->operand( in RunDotOutputFusion()
|