Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.h106 class HloShapeMatcher
109 explicit HloShapeMatcher(const Shape& shape) : shape_(shape) {} in HloShapeMatcher() function
353 return ::testing::MakeMatcher(new ::xla::testing::HloShapeMatcher(shape)); in Shape()
358 new ::xla::testing::HloShapeMatcher(ParseShape(shape).ValueOrDie())); in Shape()
Dhlo_matchers.cc152 bool HloShapeMatcher::MatchAndExplain( in MatchAndExplain()
163 void HloShapeMatcher::DescribeTo(std::ostream* os) const { in DescribeTo()