Searched refs:kPrimReluGradV2 (Results 1 – 4 of 4) sorted by relevance
32 ReLUGradV2() : PrimitiveC(prim::kPrimReluGradV2->name()) { InitIOName({"x"}, {"output"}); } in ReLUGradV2()
60 REGISTER_PRIMITIVE_EVAL_IMPL(ReLUGradV2, prim::kPrimReluGradV2, ReLUGradV2Infer, nullptr, true);
56 …VectorRef relu_grad = VectorRef({prim::kPrimReluGradV2, VectorRef({prim::kPrimAdd, x1_, x2_}), mas… in DefinePattern()
312 inline const PrimitivePtr kPrimReluGradV2 = std::make_shared<Primitive>("ReluGradV2"); variable