Searched refs:matched_layout (Results 1 – 2 of 2) sorted by relevance
41 Layout* matched_layout; in TEST() local50 .WithLayout(match::Layout(&matched_layout).WithDenseFormat())) in TEST()96 Layout* matched_layout; in TEST() local99 match::Layout(&matched_layout).WithSparseFormat()))); in TEST()102 match::Layout(&matched_layout).WithDenseFormat()))); in TEST()103 EXPECT_EQ(matched_layout, &array_shape.layout()); in TEST()119 Layout* matched_layout; in TEST() local122 match::Layout(&matched_layout).WithDenseFormat()))); in TEST()125 match::Layout(&matched_layout).WithSparseFormat()))); in TEST()126 EXPECT_EQ(matched_layout, &array_shape.layout()); in TEST()
467 LayoutType** matched_layout)468 : impl_(impl), matched_layout_(matched_layout) {}645 Layout(const ::xla::Layout** matched_layout = nullptr) {648 detail::LayoutPatternBaseImpl(), matched_layout);655 Layout(::xla::Layout** matched_layout) {657 detail::LayoutPatternBaseImpl(), matched_layout);