Searched refs:CheckTernaryShape (Results 1 – 2 of 2) sorted by relevance
109 return CheckTernaryShape(clamp); in HandleClamp()113 return CheckTernaryShape(select); in HandleSelect()117 return CheckTernaryShape(tuple_select); in HandleTupleSelect()929 Status ShapeVerifier::CheckTernaryShape(const HloInstruction* instruction) { in CheckTernaryShape() function in xla::ShapeVerifier
118 Status CheckTernaryShape(const HloInstruction* instruction);