Searched refs:HloMatcher (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 27 class HloMatcher : public ::testing::MatcherInterface<const HloInstruction*> {
29 HloMatcher(HloOpcode opcode, in HloMatcher() function
44 class HloParameterMatcher : public HloMatcher {
47 : HloMatcher(HloOpcode::kParameter, /*operands=*/{}), in HloParameterMatcher()
58 class HloComparisonMatcher : public HloMatcher {
63 : HloMatcher(HloOpcode::kCompare, operands), direction_(direction) {} in HloComparisonMatcher()
74 class HloGetTupleElementMatcher : public HloMatcher {
78 : HloMatcher(HloOpcode::kGetTupleElement, /*operands=*/{operand}), in HloGetTupleElementMatcher()
90 class HloCustomCallMatcher : public HloMatcher {
95 : HloMatcher(HloOpcode::kCustomCall, operands), in HloCustomCallMatcher()
[all …]
|
| D | hlo_matchers.cc | 25 bool HloMatcher::MatchAndExplain( in MatchAndExplain()
63 void HloMatcher::DescribeTo(::std::ostream* os) const { in DescribeTo()
80 if (!HloMatcher::MatchAndExplain(instruction, listener)) { in MatchAndExplain()
95 if (!HloMatcher::MatchAndExplain(instruction, listener)) { in MatchAndExplain()
111 if (!HloMatcher::MatchAndExplain(instruction, listener)) { in MatchAndExplain()
123 HloMatcher::DescribeTo(os); in DescribeTo()
131 if (!HloMatcher::MatchAndExplain(instruction, listener)) { in MatchAndExplain()
218 if (!HloMatcher::MatchAndExplain(instruction, listener)) { in MatchAndExplain()
245 HloMatcher::DescribeTo(os); in DescribeTo()
|