Searched refs:HloShapeAndLayoutMatcher (Results 1 – 2 of 2) sorted by relevance
119 class HloShapeAndLayoutMatcher122 explicit HloShapeAndLayoutMatcher(const Shape& shape) : shape_(shape) {} in HloShapeAndLayoutMatcher() function363 new ::xla::testing::HloShapeAndLayoutMatcher(shape)); in ShapeWithLayout()367 return ::testing::MakeMatcher(new ::xla::testing::HloShapeAndLayoutMatcher( in ShapeWithLayout()
167 bool HloShapeAndLayoutMatcher::MatchAndExplain( in MatchAndExplain()178 void HloShapeAndLayoutMatcher::DescribeTo(std::ostream* os) const { in DescribeTo()