Searched refs:incoming_loop_state (Results 1 – 1 of 1) sorted by relevance
152 const std::vector<HloInstruction*>& incoming_loop_state) { in GatherLoopBody() argument154 CHECK_EQ(incoming_loop_state.size(), 3); in GatherLoopBody()155 HloInstruction* const operand = incoming_loop_state[0]; in GatherLoopBody()156 HloInstruction* const start_indices = incoming_loop_state[1]; in GatherLoopBody()157 HloInstruction* const output_accumulator = incoming_loop_state[2]; in GatherLoopBody()