Searched refs:added_inst (Results 1 – 1 of 1) sorted by relevance
193 HloInstruction* added_inst = computation_->AddInstruction(std::move(inst)); in HandleBatchNormTraining() local194 added_inst->set_metadata(batch_norm->metadata()); in HandleBatchNormTraining()195 added_instructions.push_back(added_inst); in HandleBatchNormTraining()196 return added_inst; in HandleBatchNormTraining()361 HloInstruction* added_inst = computation_->AddInstruction(std::move(inst)); in HandleBatchNormInference() local362 added_inst->set_metadata(batch_norm->metadata()); in HandleBatchNormInference()363 added_instructions.push_back(added_inst); in HandleBatchNormInference()364 return added_inst; in HandleBatchNormInference()450 HloInstruction* added_inst = computation_->AddInstruction(std::move(inst)); in HandleBatchNormGrad() local451 added_inst->set_metadata(batch_norm->metadata()); in HandleBatchNormGrad()[all …]