Searched defs:prim_py (Results 1 – 13 of 13) sorted by relevance
132 auto prim_py = dyn_cast<PrimitivePy>(primitive); in InferShapeAndType() local
37 auto prim_py = value->cast<PrimitivePtr>(); in NeedInsertIdentity() local
70 auto prim_py = dyn_cast<PrimitivePy>(primitive); in CallPyInferFunc() local
128 auto prim_py = prim->cast<PrimitivePyPtr>(); in ClonePrim() local
199 auto prim_py = prim->cast<PrimitivePyPtr>(); in SetStrategyToCNode() local
571 auto prim_py = prim->cast<PrimitivePyPtr>(); in BpropCut() local
744 auto prim_py = value->cast_ptr<PrimitivePy>(); in GetPyObjForPrimitiveAbstract() local748 auto prim_py = prim->cast_ptr<PrimitivePy>(); in GetPyObjForPrimitiveAbstract() local1038 AbstractBasePtr PyInferRes2Abstract(const PrimitivePyPtr &prim_py, const py::dict &output) { in PyInferRes2Abstract()1065 auto prim_py = dyn_cast<PrimitivePy>(prim_); in RunPyInferValue() local1129 auto prim_py = dyn_cast<PrimitivePy>(prim_); in EvalPyCheckPrim() local2753 auto prim_py = prim->cast<PrimitivePyPtr>(); in ConvertArgsToInputs() local4694 AnfNodePtrList GetPrimitiveInitArgs(const PrimitivePyPtr &prim_py, const ops::OpDef *op_def) { in GetPrimitiveInitArgs()
1076 auto prim_py = dyn_cast<PrimitivePy>(prim); in GetPyEvaluator() local
126 auto prim_py = value->cast<PrimitivePtr>(); in UnVisited() local212 auto prim_py = value->cast<PrimitivePtr>(); in Visited() local
546 auto prim_py = prim->cast<PrimitivePyPtr>(); in BuildCustomBackwardNode() local
129 PrimitivePy::PrimitivePy(const PrimitivePy &prim_py) in PrimitivePy()
177 auto prim_py = prim->cast<PrimitivePyPtr>(); in BuildCustomBpropCNode() local
1758 auto prim_py = prim->cast<PrimitivePyPtr>(); in RunPyFunction() local2448 auto prim_py = prim->cast<PrimitivePyPtr>(); in BuildBpropCutPrim() local