Searched refs:fusion_instruction_ (Results 1 – 2 of 2) sorted by relevance
75 fusion_instruction_(fusion_instruction) {} in name_()102 HloInstruction* fusion_instruction_; variable394 bool IsFusionComputation() const { return fusion_instruction_ != nullptr; } in IsFusionComputation()398 HloInstruction* FusionInstruction() const { return fusion_instruction_; } in FusionInstruction()400 fusion_instruction_ = fusion_instruction; in SetFusionInstruction()460 HloInstruction* fusion_instruction_; variable
64 name_, parameter_count, &instructions_, root, fusion_instruction_)); in Build()74 fusion_instruction_(fusion_instruction) { in HloComputation()120 CHECK(fusion_instruction_->operand_count() == param_instructions_.size()); in AddParameter()