Searched refs:kPrimSoftplus (Results 1 – 4 of 4) sorted by relevance
34 Softplus() : PrimitiveC(prim::kPrimSoftplus->name()) { InitIOName({"x"}, {"output"}); } in Softplus()
54 REGISTER_PRIMITIVE_EVAL_IMPL(Softplus, prim::kPrimSoftplus, SoftplusInfer, nullptr, true);
48 {prim::kPrimSoftplus->name(), DescParam{dnnl::algorithm::eltwise_soft_relu}}, in GetForwardEltwiseDesc()
372 inline const PrimitivePtr kPrimSoftplus = std::make_shared<Primitive>("Softplus"); variable