Searched refs:shape_c (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1567 const auto shape_c = properties.GetOutputProperties("c").at(0).shape(); in TEST_F() local
1569 EXPECT_EQ(shape_a.dim_size(), shape_c.dim_size()); in TEST_F()
1571 EXPECT_EQ(shape_a.dim(0).size(), shape_c.dim(0).size()); in TEST_F()
1573 EXPECT_EQ(shape_a.dim(1).size(), shape_c.dim(1).size()); in TEST_F()
1589 EXPECT_EQ(shape_e.dim(0).size(), shape_c.dim(0).size()); in TEST_F()
1590 EXPECT_NE(shape_e.dim(1).size(), shape_c.dim(1).size()); in TEST_F()
1600 EXPECT_EQ(shape_h.dim(0).size(), shape_c.dim(0).size()); in TEST_F()
1601 EXPECT_EQ(shape_h.dim(1).size(), shape_c.dim(1).size()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 193 Shape shape_c = ShapeUtil::MakeShape(F32, {4, 3}, {false, true}); in TEST() local
194 *shape_c.mutable_layout() = Layout({0, 1}); in TEST()
198 EXPECT_FALSE(ShapeUtil::Compatible(shape_a, shape_c)); in TEST()
|