Home
last modified time | relevance | path

Searched refs:IsArrayPrimitiveType (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc83 /* static */ bool ShapeUtil::IsArrayPrimitiveType( in IsArrayPrimitiveType() function in xla::ShapeUtil
172 CHECK(IsArrayPrimitiveType(element_type)) << element_type; in MakeValidatedShape()
208 CHECK(IsArrayPrimitiveType(element_type)); in MakeShapeWithSparseLayout()
Dshape_util.h458 static bool IsArrayPrimitiveType(PrimitiveType primitive_type);
Dliteral_test.cc1449 CHECK(ShapeUtil::IsArrayPrimitiveType(shape_proto->element_type())); in SetDefaultLayoutOnProto()