Searched refs:mutable_instruction (Results 1 – 11 of 11) sorted by relevance
603 HloInstruction* mutable_instruction = in AcceptOrdered() local605 TF_RETURN_IF_ERROR(visitor->Preprocess(mutable_instruction)); in AcceptOrdered()606 TF_RETURN_IF_ERROR(mutable_instruction->Visit(visitor)); in AcceptOrdered()607 visitor->SetVisited(*mutable_instruction); in AcceptOrdered()608 TF_RETURN_IF_ERROR(visitor->Postprocess(mutable_instruction)); in AcceptOrdered()
186 *wrapper_info.mutable_instruction() = in Apply()
46 *message_.mutable_instruction() = instruction; in TransformationMoveInstructionDown()
189 *wrapper_info.mutable_instruction() = in Apply()
580 *instr_log.mutable_instruction() = instr->ToProto(); in PickBestAlgorithmNoCacheCuda()
33 *result.mutable_instruction() = instruction; in MakeSideEffectWrapperInfo()