Searched refs:CheckOperandCount (Results 1 – 1 of 1) sorted by relevance
63 Status CheckOperandCount(const HloInstruction* hlo, int expected) { in CheckOperandCount() function95 TF_RETURN_IF_ERROR(CheckOperandCount(hlo, *arity)); in Preprocess()181 TF_RETURN_IF_ERROR(CheckOperandCount(hlo, 1)); in HandleCholesky()283 TF_RETURN_IF_ERROR(CheckOperandCount(instruction, 2)); in HandleRng()676 TF_RETURN_IF_ERROR(CheckOperandCount(conditional, num_branches + 1)); in HandleConditional()