Searched refs:bshape_with_layout (Results 1 – 1 of 1) sorted by relevance
367 Shape bshape_with_layout(bshape); in TEST_F() local369 *bshape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()374 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()408 Shape bshape_with_layout(bshape); in TEST_F() local410 *bshape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({0, 1}); in TEST_F()415 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()420 EXPECT_TRUE(LayoutUtil::Equal(bshape_with_layout.layout(), in TEST_F()423 LayoutUtil::Equal(bshape_with_layout.layout(), tanh->shape().layout())); in TEST_F()