Searched refs:param_instruction (Results 1 – 1 of 1) sorted by relevance
172 HloInstruction* param_instruction = param_instructions_[param_no]; in RemoveParameter() local176 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveParameter()179 param_instruction = param_instructions_[param_no]; in RemoveParameter()182 param_no, param_instruction->shape(), StrCat("param_", param_no))); in RemoveParameter()183 TF_RETURN_IF_ERROR(param_instruction->ReplaceAllUsesWith(new_instr)); in RemoveParameter()185 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveParameter()204 HloInstruction* param_instruction = param_instructions_[i]; in RemoveUnusedParametersImpl() local205 if (param_instruction->user_count() == 0 && in RemoveUnusedParametersImpl()206 param_instruction != root_instruction()) { in RemoveUnusedParametersImpl()208 RemoveInstructionImpl(param_instruction, allow_non_fusion)); in RemoveUnusedParametersImpl()[all …]