Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher_test.cc43 Shape* matched_shape; in TEST_F() local
52 match::Shape(&matched_shape) in TEST_F()
73 Shape* matched_shape; in TEST_F() local
74 EXPECT_TRUE(Match(&scalar_shape, match::Shape(&matched_shape).IsScalar())); in TEST_F()
75 EXPECT_EQ(matched_shape, &scalar_shape); in TEST_F()
88 Shape* matched_shape; in TEST_F() local
89 EXPECT_TRUE(Match(&array_shape, match::Shape(&matched_shape).IsArray())); in TEST_F()
90 EXPECT_EQ(matched_shape, &array_shape); in TEST_F()
Dpattern_matcher.h936 explicit constexpr ShapePattern(const Impl& impl, ShapeType** matched_shape)
937 : impl_(impl), matched_shape_(matched_shape) {}
1074 inline constexpr auto Shape(const ::xla::Shape** matched_shape = nullptr) {
1076 detail::ShapePatternBaseImpl(), matched_shape);
1080 inline constexpr auto Shape(::xla::Shape** matched_shape) {
1082 detail::ShapePatternBaseImpl(), matched_shape);