Searched refs:HloShapeMatcher (Results 1 – 2 of 2) sorted by relevance
106 class HloShapeMatcher109 explicit HloShapeMatcher(const Shape& shape) : shape_(shape) {} in HloShapeMatcher() function353 return ::testing::MakeMatcher(new ::xla::testing::HloShapeMatcher(shape)); in Shape()358 new ::xla::testing::HloShapeMatcher(ParseShape(shape).ValueOrDie())); in Shape()
152 bool HloShapeMatcher::MatchAndExplain( in MatchAndExplain()163 void HloShapeMatcher::DescribeTo(std::ostream* os) const { in DescribeTo()