Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.h419 static bool IsArrayPrimitiveType(PrimitiveType primitive_type);
H A Dshape_util.cc91 /* static */ bool ShapeUtil::IsArrayPrimitiveType( in IsArrayPrimitiveType() function in xla::ShapeUtil
H A Dliteral_test.cc1527 CHECK(ShapeUtil::IsArrayPrimitiveType(shape_proto->element_type())); in SetDefaultLayoutOnProto()