Searched refs:HasPrimitiveType (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 344 TEST(ShapeUtilTest, HasPrimitiveType) { in TEST() argument
345 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {}), S32)); in TEST()
346 EXPECT_FALSE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {}), S16)); in TEST()
347 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {0}), S32)); in TEST()
348 EXPECT_FALSE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeTupleShape({}), S32)); in TEST()
349 EXPECT_TRUE(ShapeUtil::HasPrimitiveType( in TEST()
353 EXPECT_TRUE(ShapeUtil::HasPrimitiveType( in TEST()
|
| D | shape_util.cc | 401 /* static */ bool ShapeUtil::HasPrimitiveType(const Shape& shape, in HasPrimitiveType() function in xla::ShapeUtil
407 if (HasPrimitiveType(element_shape, primitive_type)) { in HasPrimitiveType()
|
| D | shape_util.h | 195 static bool HasPrimitiveType(const Shape& shape,
|
| D | literal.cc | 296 if (ShapeUtil::HasPrimitiveType(shape, OPAQUE)) { in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_get_dimension_size_rewriter.cc | 41 TF_RET_CHECK(ShapeUtil::HasPrimitiveType(instr->shape(), U32)); in ReplaceGetSize()
|
| D | hlo_graph_dumper.cc | 123 if (ShapeUtil::HasPrimitiveType(instr->shape(), OPAQUE) || in IsSmall()
124 ShapeUtil::HasPrimitiveType(instr->shape(), TOKEN)) { in IsSmall()
|
| D | algebraic_simplifier.cc | 570 if (ShapeUtil::HasPrimitiveType(lhs->shape(), xla::PRED) && in HandleAnd()
571 ShapeUtil::HasPrimitiveType(rhs->shape(), xla::PRED)) { in HandleAnd()
1884 if (ShapeUtil::HasPrimitiveType(lhs->shape(), xla::PRED) && in HandleOr()
1885 ShapeUtil::HasPrimitiveType(rhs->shape(), xla::PRED)) { in HandleOr()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 1234 EXPECT_TRUE(xla::ShapeUtil::HasPrimitiveType( in TEST()
|