Searched refs:async_instruction (Results 1 – 1 of 1) sorted by relevance
603 void AddAsyncInstruction(HloInstruction* async_instruction) { in AddAsyncInstruction() argument604 CHECK(async_instruction != nullptr) in AddAsyncInstruction()607 CHECK(async_instruction->opcode() == HloOpcode::kAsyncStart || in AddAsyncInstruction()608 async_instruction->opcode() == HloOpcode::kAsyncUpdate || in AddAsyncInstruction()609 async_instruction->opcode() == HloOpcode::kAsyncDone); in AddAsyncInstruction()610 async_instructions_.push_back(async_instruction); in AddAsyncInstruction()613 void RemoveAsyncInstruction(HloInstruction* async_instruction) { in RemoveAsyncInstruction() argument614 if (async_instruction == nullptr) { in RemoveAsyncInstruction()619 async_instruction), in RemoveAsyncInstruction()