| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_get_dimension_size_rewriter.cc | 41 TF_RET_CHECK(ShapeUtil::HasPrimitiveType(instr->shape(), U32));
|
| hlo_graph_dumper.cc | 123 if (ShapeUtil::HasPrimitiveType(instr->shape(), OPAQUE) ||
124 ShapeUtil::HasPrimitiveType(instr->shape(), TOKEN)) {
[all...] |
| algebraic_simplifier.cc | 570 if (ShapeUtil::HasPrimitiveType(lhs->shape(), xla::PRED) &&
571 ShapeUtil::HasPrimitiveType(rhs->shape(), xla::PRED)) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util_test.cc | 344 TEST(ShapeUtilTest, HasPrimitiveType) {
345 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {}), S32));
346 EXPECT_FALSE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {}), S16));
347 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeShape(S32, {0}), S32));
348 EXPECT_FALSE(ShapeUtil::HasPrimitiveType(ShapeUtil::MakeTupleShape({}), S32));
349 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(
353 EXPECT_TRUE(ShapeUtil::HasPrimitiveType(
[all...] |
| shape_util.h | 195 static bool HasPrimitiveType(const Shape& shape,
[all...] |
| shape_util.cc | 401 /* static */ bool ShapeUtil::HasPrimitiveType(const Shape& shape,
407 if (HasPrimitiveType(element_shape, primitive_type)) {
[all...] |
| literal.cc | 296 if (ShapeUtil::HasPrimitiveType(shape, OPAQUE)) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| raw_api_test.cc | [all...] |
