Searched refs:kPrimGeLUGrad (Results 1 – 4 of 4) sorted by relevance
144 REG_ADPT_DESC(GeluGrad, prim::kPrimGeLUGrad->name(), ADPT_DESC(GeluGrad))
58 {kAllTarget, OpLevel_0, prim::kPrimGeLUGrad}, in GetExpandOps()
242 {prim::kPrimGeLUGrad->name(), &EltWiseGradCPUKernel<T>::GeluGrad}, in Launch()
364 inline const PrimitivePtr kPrimGeLUGrad = std::make_shared<Primitive>(kGeLUGrad); variable