Lines Matching refs:kWhile
283 // The shape of kWhile should match the shape of the body computation it398 case HloOpcode::kWhile:478 case HloOpcode::kWhile:770 } else if (instruction->opcode() == HloOpcode::kWhile) {