Searched refs:CheckUnaryShape (Results 1 – 2 of 2) sorted by relevance
116 Status CheckUnaryShape(const HloInstruction* instruction);
101 return CheckUnaryShape(hlo); in HandleElementwiseUnary()143 return CheckUnaryShape(copy); in HandleCopy()916 Status ShapeVerifier::CheckUnaryShape(const HloInstruction* instruction) { in CheckUnaryShape() function in xla::ShapeVerifier