Searched refs:calling_instruction (Results 1 – 2 of 2) sorted by relevance
1226 const HloInstruction* calling_instruction = in IsWhileBody() local1228 if (calling_instruction->opcode() == HloOpcode::kWhile && in IsWhileBody()1229 calling_instruction->while_body() == node.computation()) { in IsWhileBody()
72 Status CheckParameterCount(const HloInstruction* calling_instruction, in CheckParameterCount() argument77 computation->name(), calling_instruction->name(), expected, in CheckParameterCount()