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.cc40 Shape* matched_shape; in TEST() local
49 match::Shape(&matched_shape) in TEST()
70 Shape* matched_shape; in TEST() local
71 EXPECT_TRUE(Match(&scalar_shape, match::Shape(&matched_shape).IsScalar())); in TEST()
72 EXPECT_EQ(matched_shape, &scalar_shape); in TEST()
85 Shape* matched_shape; in TEST() local
86 EXPECT_TRUE(Match(&array_shape, match::Shape(&matched_shape).IsArray())); in TEST()
87 EXPECT_EQ(matched_shape, &array_shape); in TEST()
108 Shape* matched_shape; in TEST() local
109 EXPECT_TRUE(Match(&array_shape, match::Shape(&matched_shape).IsArray())); in TEST()
[all …]
Dpattern_matcher.h957 explicit constexpr ShapePattern(const Impl& impl, ShapeType** matched_shape)
958 : impl_(impl), matched_shape_(matched_shape) {}
1116 Shape(const ::xla::Shape** matched_shape = nullptr) {
1118 detail::ShapePatternBaseImpl(), matched_shape);
1124 Shape(::xla::Shape** matched_shape) {
1126 detail::ShapePatternBaseImpl(), matched_shape);