Searched refs:matched_inst (Results 1 – 2 of 2) sorted by relevance
1801 HloInstructionType** matched_inst)1802 : impl_(impl), matched_inst_(matched_inst) {}2017 Op(const ::xla::HloInstruction** matched_inst = nullptr) {2020 detail::HloInstructionPatternBaseImpl(), matched_inst);2027 Op(::xla::HloInstruction** matched_inst) {2030 detail::HloInstructionPatternBaseImpl(), matched_inst);2040 inline auto NAME(HloInstructionType** matched_inst) \2041 ->decltype(Op(matched_inst).WithOpcode(HloOpcode::k##NAME)) { \2042 return Op(matched_inst).WithOpcode(HloOpcode::k##NAME); \2066 inline auto NAME(HloInstructionType** matched_inst, Arg&& arg) \[all …]
38 const HloInstruction* matched_inst; in TEST() local45 match::Op(&matched_inst) in TEST()54 ASSERT_NE(matched_inst, nullptr); in TEST()55 EXPECT_EQ(matched_inst->name(), "two_plus_two"); in TEST()56 EXPECT_EQ(matched_inst->opcode(), HloOpcode::kAdd); in TEST()63 matched_inst = nullptr; in TEST()65 match::Multiply(&matched_inst, match::Op(), match::Op()))); in TEST()