Searched refs:shape_o2 (Results 1 – 1 of 1) sorted by relevance
1653 const auto shape_o2 = properties.GetOutputProperties("o2").at(0).shape(); in TEST_F() local1655 EXPECT_EQ(shape_b.DebugString(), shape_o2.DebugString()); in TEST_F()1758 const auto shape_o2 = properties.GetOutputProperties("o2").at(0).shape(); in TEST_F() local1761 EXPECT_EQ(1, shape_o2.dim_size()); in TEST_F()1763 EXPECT_EQ(shape_a.dim(1).size(), shape_o2.dim(0).size()); in TEST_F()