Searched refs:IsArrayPrimitiveType (Results 1 – 3 of 3) sorted by relevance
83 /* static */ bool ShapeUtil::IsArrayPrimitiveType( in IsArrayPrimitiveType() function in xla::ShapeUtil199 CHECK(IsArrayPrimitiveType(element_type)) << element_type; in MakeValidatedShape()
471 static bool IsArrayPrimitiveType(PrimitiveType primitive_type);
1412 CHECK(ShapeUtil::IsArrayPrimitiveType(shape_proto->element_type())); in SetDefaultLayoutOnProto()