Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.h135 class HloShardingMatcher
138 explicit HloShardingMatcher(const absl::optional<HloSharding>& sharding) in HloShardingMatcher() function
375 new ::xla::testing::HloShardingMatcher(sharding)); in Sharding()
380 return ::testing::MakeMatcher(new ::xla::testing::HloShardingMatcher( in Sharding()
386 new ::xla::testing::HloShardingMatcher(absl::nullopt)); in NoSharding()
Dhlo_matchers.cc182 bool HloShardingMatcher::MatchAndExplain( in MatchAndExplain()
207 void HloShardingMatcher::DescribeTo(std::ostream* os) const { in DescribeTo()