Searched refs:CheckParameterCount (Results 1 – 1 of 1) sorted by relevance
72 Status CheckParameterCount(const HloInstruction* calling_instruction, in CheckParameterCount() function356 CheckParameterCount(sort, compare, sort->operand_count() * 2)); in HandleSort()536 CheckParameterCount(call, call->to_apply(), call->operand_count())); in HandleCall()648 CheckParameterCount(xla_while, xla_while->while_body(), 1)); in HandleWhile()650 CheckParameterCount(xla_while, xla_while->while_condition(), 1)); in HandleWhile()678 TF_RETURN_IF_ERROR(CheckParameterCount( in HandleConditional()