Searched refs:col_major_layout (Results 1 – 1 of 1) sorted by relevance
136 Layout col_major_layout = LayoutUtil::MakeLayout({1, 0}); in TEST_F() local138 *col_major_shape.mutable_layout() = col_major_layout; in TEST_F()152 EXPECT_TRUE(LayoutUtil::Equal(col_major_layout, param0->shape().layout())); in TEST_F()155 col_major_layout, computation->root_instruction()->shape().layout())); in TEST_F()