Searched refs:CheckParameterCount (Results 1 – 1 of 1) sorted by relevance
69 Status CheckParameterCount(const HloInstruction* calling_instruction, in CheckParameterCount() function561 CheckParameterCount(sort, compare, sort->operand_count() * 2)); in HandleSort()784 CheckParameterCount(call, call->to_apply(), call->operand_count())); in HandleCall()922 CheckParameterCount(xla_while, xla_while->while_body(), 1)); in HandleWhile()924 CheckParameterCount(xla_while, xla_while->while_condition(), 1)); in HandleWhile()966 TF_RETURN_IF_ERROR(CheckParameterCount( in HandleConditional()