Searched refs:matched_inst (Results 1 – 2 of 2) sorted by relevance
1774 HloInstructionType** matched_inst)1775 : impl_(impl), matched_inst_(matched_inst) {}1959 inline constexpr auto Op(const ::xla::HloInstruction** matched_inst = nullptr) {1962 detail::HloInstructionPatternBaseImpl(), matched_inst);1967 inline constexpr auto Op(::xla::HloInstruction** matched_inst) {1970 detail::HloInstructionPatternBaseImpl(), matched_inst);1978 inline auto NAME(HloInstructionType** matched_inst) { \1979 return Op(matched_inst).WithOpcode(HloOpcode::k##NAME); \1999 inline auto NAME(HloInstructionType** matched_inst, Arg&& arg) { \2000 return Op(matched_inst) \[all …]
41 const HloInstruction* matched_inst; in TEST_F() local48 match::Op(&matched_inst) in TEST_F()57 ASSERT_NE(matched_inst, nullptr); in TEST_F()58 EXPECT_EQ(matched_inst->name(), "two_plus_two"); in TEST_F()59 EXPECT_EQ(matched_inst->opcode(), HloOpcode::kAdd); in TEST_F()66 matched_inst = nullptr; in TEST_F()68 match::Multiply(&matched_inst, match::Op(), match::Op()))); in TEST_F()