Searched refs:PrimitivePyPtr (Results 1 – 16 of 16) sorted by relevance
39 using PrimitivePyPtr = std::shared_ptr<PrimitivePy>; variable102 void set_attached_primitive(const PrimitivePyPtr &prim);103 PrimitivePyPtr attached_primitive() { return attached_primitive_.lock(); } in attached_primitive()
316 auto primitive_py = primitive->cast<PrimitivePyPtr>(); in CopyHookFunction()496 void PrimitivePyAdapter::set_attached_primitive(const PrimitivePyPtr &prim) { in set_attached_primitive()
108 PrimitivePyPtr matched_primitive() { return matched_prim_; } in matched_primitive()116 vector<PrimitivePyPtr> primitives_;118 PrimitivePyPtr matched_prim_{nullptr};
62 PrimitivePyPtr prim = prim_pattern->matched_primitive(); in BuildPrimitive()
56 explicit PythonPrimEvaluator(const PrimitivePyPtr primitive) in PythonPrimEvaluator()66 PrimitivePyPtr prim_py_;
283 std::unordered_map<PrimitivePyPtr, EvaluatorPtr> prim_py_evaluators_;
514 py::tuple PreparePyInputs(const PrimitivePyPtr &, const AbstractBasePtrList &args) { in PreparePyInputs() argument549 AbstractBasePtr PyInferRes2Abstract(const PrimitivePyPtr &prim_py, const py::dict &output) { in PyInferRes2Abstract()
59 PrimitivePyPtr py_primitive;
354 …void DoSignatrueCast(const PrimitivePyPtr &prim, const std::unordered_map<SignatureEnumDType, Type…
276 bool GetSignatureType(const PrimitivePyPtr &prim, std::vector<SignatureEnumDType> *dtypes) { in GetSignatureType()292 void PynativeInfer(const PrimitivePyPtr &prim, OpExecInfo *const op_exec_info, in PynativeInfer()1211 void ForwardExecutor::DoSignatrueCast(const PrimitivePyPtr &prim, in DoSignatrueCast()
238 py::function fn = prim->cast<PrimitivePyPtr>()->GetBpropFunction(); in ExportBpropMindir()281 fn = prim->cast<PrimitivePyPtr>()->GetBpropFunction(); in GetBprop()
40 return function->cast<PrimitivePyPtr>()->signatures(); in GetSignature()
425 PrimitivePyPtr conv2d_ptr = dyn_cast<PrimitivePy>(conv2d_prim); in TEST_F()
287 PrimitivePyPtr allreduce_ptr = dyn_cast<PrimitivePy>(op_instance); in TEST_F()
178 auto cast_py = cast_node->cast<ValueNodePtr>()->value()->cast<PrimitivePyPtr>(); in TEST_F()
587 auto quant_op = quant_op_value->cast<PrimitivePyPtr>(); in GetWeightInfo()