Searched refs:param_instruction (Results 1 – 1 of 1) sorted by relevance
149 HloInstruction* param_instruction = param_instructions_[param_no]; in RemoveParameter() local153 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveParameter()156 param_instruction = param_instructions_[param_no]; in RemoveParameter()159 param_no, param_instruction->shape(), StrCat("param_", param_no))); in RemoveParameter()160 TF_RETURN_IF_ERROR(param_instruction->ReplaceAllUsesWith(new_instr)); in RemoveParameter()162 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveParameter()173 HloInstruction* param_instruction = param_instructions_[i]; in RemoveUnusedParameters() local174 if (param_instruction->user_count() == 0 && in RemoveUnusedParameters()175 param_instruction != root_instruction()) { in RemoveUnusedParameters()176 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveUnusedParameters()[all …]