| /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 | 542 HloInstruction* fusion_instruction = nullptr; in Run() local
549 fusion_instruction = Fuse(operand, instruction); in Run()
555 fusion_instruction = FuseIntoMultiOutput(operand, instruction); in Run()
559 if (fusion_instruction == nullptr) { in Run()
564 fusion_queue->OnFusingInstruction(fusion_instruction, operand, in Run()
577 if (fusion_instruction != instruction) { in Run()
622 HloInstruction* fusion_instruction; in AddFusionInstruction() local
625 fusion_instruction = consumer; in AddFusionInstruction()
626 if (kind != fusion_instruction->fusion_kind()) { in AddFusionInstruction()
627 fusion_instruction->set_fusion_kind(kind); in AddFusionInstruction()
[all …]
|
| D | fusion_node_indexing_evaluation_test.cc | 42 HloInstruction* FuseInstruction(HloInstruction* fusion_instruction, in FuseInstruction() argument
44 auto evaluation = fusion_node_evaluations_.find(fusion_instruction); in FuseInstruction()
48 .emplace(fusion_instruction, in FuseInstruction()
49 FusionNodeIndexingEvaluation(fusion_instruction)) in FuseInstruction()
54 InstructionFusion::FuseInstruction(fusion_instruction, producer); in FuseInstruction()
|
| D | hlo_computation.h | 71 HloInstruction* fusion_instruction = nullptr)
74 fusion_instruction_(fusion_instruction) {} in name_()
465 void SetFusionInstruction(HloInstruction* fusion_instruction) { in SetFusionInstruction() argument
466 fusion_instruction_ = fusion_instruction; in SetFusionInstruction()
499 HloInstruction* root_instruction, HloInstruction* fusion_instruction);
514 HloInstruction* fusion_instruction);
|
| D | hlo_computation.cc | 70 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation() argument
74 fusion_instruction_(fusion_instruction) { in HloComputation()
725 HloInstruction* fusion_instruction) { in FuseInstructionsInto() argument
726 CHECK_EQ(HloOpcode::kFusion, fusion_instruction->opcode()); in FuseInstructionsInto()
728 TF_CHECK_OK(root->ReplaceAllUsesWith(fusion_instruction)); in FuseInstructionsInto()
730 set_root_instruction(fusion_instruction); in FuseInstructionsInto()
735 fusion_instruction->FuseInstruction(instruction); in FuseInstructionsInto()
746 HloInstruction* fusion_instruction = AddInstruction( in CreateFusionInstruction() local
748 FuseInstructionsInto(instructions_to_fuse, fusion_instruction); in CreateFusionInstruction()
749 return fusion_instruction; in CreateFusionInstruction()
|
| D | instruction_fusion.h | 97 virtual HloInstruction* FuseInstruction(HloInstruction* fusion_instruction,
|
| D | memory_space_assignment_test.cc | 4770 HloInstruction* fusion_instruction = in TEST_P() local
4773 GetAlternateMemoryOffset(*preset_assignments, fusion_instruction); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | instruction_fusion.cc | 134 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction() argument
135 auto evaluation = fusion_node_evaluations_.find(fusion_instruction); in FuseInstruction()
138 .emplace(fusion_instruction, in FuseInstruction()
139 FusionNodeIndexingEvaluation(fusion_instruction)) in FuseInstruction()
144 InstructionFusion::FuseInstruction(fusion_instruction, producer); in FuseInstruction()
|
| D | instruction_fusion.h | 53 HloInstruction* FuseInstruction(HloInstruction* fusion_instruction,
|
| /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_instruction_fusion.cc | 206 HloInstruction* fusion_instruction, HloInstruction* producer) { in FuseInstruction() argument
207 auto evaluation = fusion_node_evaluations_.find(fusion_instruction); in FuseInstruction()
210 .emplace(fusion_instruction, in FuseInstruction()
211 FusionNodeIndexingEvaluation(fusion_instruction)) in FuseInstruction()
216 InstructionFusion::FuseInstruction(fusion_instruction, producer); in FuseInstruction()
|
| D | cpu_layout_assignment_test.cc | 300 HloInstruction* fusion_instruction = in RunDotOutputFusion() local
304 computation->ReplaceInstruction(add_result, fusion_instruction)); in RunDotOutputFusion()
307 fusion_instruction->fused_instructions_computation()->root_instruction(); in RunDotOutputFusion()
308 HloInstruction* fused_dot = fusion_instruction->FuseInstruction(dot_result); in RunDotOutputFusion()
322 fusion_instruction->operand(fused_dot->operand(0)->parameter_number()); in RunDotOutputFusion()
324 fusion_instruction->operand(fused_dot->operand(1)->parameter_number()); in RunDotOutputFusion()
325 result.addend_fusion_param = fusion_instruction->operand( in RunDotOutputFusion()
|
| D | cpu_instruction_fusion.h | 44 HloInstruction* FuseInstruction(HloInstruction* fusion_instruction,
|