Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.h119 class HloShapeAndLayoutMatcher
122 explicit HloShapeAndLayoutMatcher(const Shape& shape) : shape_(shape) {} in HloShapeAndLayoutMatcher() function
363 new ::xla::testing::HloShapeAndLayoutMatcher(shape)); in ShapeWithLayout()
367 return ::testing::MakeMatcher(new ::xla::testing::HloShapeAndLayoutMatcher( in ShapeWithLayout()
Dhlo_matchers.cc167 bool HloShapeAndLayoutMatcher::MatchAndExplain( in MatchAndExplain()
178 void HloShapeAndLayoutMatcher::DescribeTo(std::ostream* os) const { in DescribeTo()