Home
last modified time | relevance | path

Searched refs:matched_layout (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher_test.cc44 Layout* matched_layout; in TEST_F() local
53 .WithLayout(match::Layout(&matched_layout).WithDenseFormat())) in TEST_F()
98 Layout* matched_layout; in TEST_F() local
101 match::Layout(&matched_layout).WithDenseFormat()))); in TEST_F()
102 EXPECT_EQ(matched_layout, &array_shape.layout()); in TEST_F()
Dpattern_matcher.h461 LayoutType** matched_layout)
462 : impl_(impl), matched_layout_(matched_layout) {}
628 inline constexpr auto Layout(const ::xla::Layout** matched_layout = nullptr) {
631 detail::LayoutPatternBaseImpl(), matched_layout);
636 inline constexpr auto Layout(::xla::Layout** matched_layout) {
638 detail::LayoutPatternBaseImpl(), matched_layout);