Searched refs:kPrimCdistGrad (Results 1 – 3 of 3) sorted by relevance
92 auto cdist_grad_prim = std::make_shared<Primitive>(prim::kPrimCdistGrad->name()); in DefinePattern()147 …std::vector<AnfNodePtr> new_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimCdistGrad->… in Process()
70 REGISTER_PRIMITIVE_EVAL_IMPL(CdistGrad, prim::kPrimCdistGrad, CdistGradInfer, nullptr, true);
524 inline const PrimitivePtr kPrimCdistGrad = std::make_shared<Primitive>(kCdistGrad); variable