Searched refs:next_operand (Results 1 – 3 of 3) sorted by relevance
90 HloInstruction* next_operand = in CreateLoopInvariantCopy() local92 if (hoisted_instructions->contains(next_operand) || in CreateLoopInvariantCopy()93 next_operand == while_body_param) { in CreateLoopInvariantCopy()97 dfs_stack.push_back({next_operand, 0}); in CreateLoopInvariantCopy()
102 HloInstruction* next_operand = in CreateLoopInvariantCopy() local104 if (next_operand == while_body_param || in CreateLoopInvariantCopy()105 FindOrDie(*invariant_instructions, next_operand).hoisted_copy != in CreateLoopInvariantCopy()110 dfs_stack.push_back({next_operand, 0}); in CreateLoopInvariantCopy()
3650 int64 next_operand = item.processed_operands.size(); in BuildDynamicInferenceGraph() local3659 next_operand++; in BuildDynamicInferenceGraph()3696 if (opcode == HloOpcode::kGetDimensionSize && next_operand == 0) { in BuildDynamicInferenceGraph()3700 if (next_operand >= instr_proto->operand_ids_size() || in BuildDynamicInferenceGraph()3715 WorkItem next_item(instr_proto->operand_ids(next_operand), in BuildDynamicInferenceGraph()3717 if (opcode == HloOpcode::kSelect && next_operand == 0) { in BuildDynamicInferenceGraph()3720 if (opcode == HloOpcode::kGather && next_operand == 1) { in BuildDynamicInferenceGraph()