Searched refs:ShapeIs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | tuple_test.cc | 173 ASSERT_TRUE(ShapeUtil::ShapeIs(*vector_shape, F32, {3})); in XLA_TEST_F()
174 ASSERT_TRUE(ShapeUtil::ShapeIs(*matrix_shape, F32, {/*y=*/2, /*x=*/3})); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 559 TEST(ShapeUtilTest, ShapeIs) { in TEST() argument
560 EXPECT_FALSE(ShapeUtil::ShapeIs(ShapeUtil::MakeShape(PRED, {2}), PRED, {})); in TEST()
|
| D | shape_util.h | 430 static bool ShapeIs(const Shape& shape, PrimitiveType element_type,
|
| D | shape_util.cc | 366 /* static */ bool ShapeUtil::ShapeIs(const Shape& shape, in ShapeIs() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2201 if (!ShapeUtil::ShapeIs(condition.result(), PRED, {})) { in InferWhileShape()
2221 if (!ShapeUtil::ShapeIs(predicate, PRED, {})) { in InferConditionalShape()
|