Searched refs:fusion_instruction_ (Results 1 – 3 of 3) sorted by relevance
77 fusion_instruction_(fusion_instruction) {} in name_()114 HloInstruction* fusion_instruction_; variable569 HloInstruction* FusionInstruction() const { return fusion_instruction_; } in FusionInstruction()572 fusion_instruction_ = fusion_instruction; in SetFusionInstruction()714 HloInstruction* fusion_instruction_; variable
72 name_, parameter_count, &instructions_, root, fusion_instruction_)); in Build()82 fusion_instruction_(fusion_instruction), in HloComputation()107 if (fusion_instruction_ != nullptr) { in ~HloComputation()108 CHECK(fusion_instruction_->fused_instructions_computation() == this); in ~HloComputation()109 fusion_instruction_->ClearCalledComputations(); in ~HloComputation()110 fusion_instruction_ = nullptr; in ~HloComputation()157 fusion_instruction_->operand_count() == param_instructions_.size()); in AddParameter()231 fusion_instruction_->operand_count() == param_instructions_.size()); in ReplaceParameter()
365 // parent fusion_instruction_ may get removed and be nullptr.