Searched refs:HasPrimitiveType (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 368 TEST(ShapeUtilTest, HasPrimitiveType) { in TEST() argument
369 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {}), S32)); in TEST()
370 EXPECT_FALSE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {}), S16)); in TEST()
371 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {0}), S32)); in TEST()
372 EXPECT_FALSE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeTupleShape({}), S32)); in TEST()
373 EXPECT_TRUE(ShapeUtil::HasPrimitiveType( in TEST()
377 EXPECT_TRUE(ShapeUtil::HasPrimitiveType( in TEST()
|
| D | shape_util.h | 204 static bool HasPrimitiveType(const Shape& shape,
|
| D | shape_util.cc | 558 /* static */ bool ShapeUtil::HasPrimitiveType(const Shape& shape, in HasPrimitiveType() function in xla::ShapeUtil
564 if (HasPrimitiveType(element_shape, primitive_type)) { in HasPrimitiveType()
|
| D | literal.cc | 356 if (ShapeUtil::HasPrimitiveType(shape, OPAQUE_TYPE)) { in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 122 if (ShapeUtil::HasPrimitiveType(instr->shape(), OPAQUE_TYPE) || in IsSmall()
123 ShapeUtil::HasPrimitiveType(instr->shape(), TOKEN)) { in IsSmall()
|
| D | algebraic_simplifier.cc | 1118 if (ShapeUtil::HasPrimitiveType(lhs->shape(), xla::PRED) && in HandleAnd()
1119 ShapeUtil::HasPrimitiveType(rhs->shape(), xla::PRED)) { in HandleAnd()
2905 if (ShapeUtil::HasPrimitiveType(lhs->shape(), xla::PRED) && in HandleOr()
2906 ShapeUtil::HasPrimitiveType(rhs->shape(), xla::PRED)) { in HandleOr()
|
| D | dynamic_padder.cc | 167 TF_RET_CHECK(ShapeUtil::HasPrimitiveType(instr->shape(), S32)); in ReplaceGetSize()
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 2036 EXPECT_TRUE(xla::ShapeUtil::HasPrimitiveType( in TEST()
|