Searched refs:prim_adapter (Results 1 – 4 of 4) sorted by relevance
94 auto prim_adapter = prim_obj.cast<PrimitivePyAdapterPtr>(); in Prim() local95 primitives_.push_back(std::make_shared<PrimitivePy>(prim_obj, prim_adapter)); in Prim()133 auto prim_adapter = prim_obj.cast<PrimitivePyAdapterPtr>(); in Call() local134 prim_ = std::make_shared<PrimitivePy>(prim_obj, prim_adapter); in Call()
97 auto prim_adapter = ops_list[i].cast<PrimitivePyAdapterPtr>(); in GetSerializableBpropList() local98 if (prim_adapter == nullptr) { in GetSerializableBpropList()102 (void)serializable_bprop_list.insert(prim_adapter->name()); in GetSerializableBpropList()226 auto prim_adapter = obj.cast<PrimitivePyAdapterPtr>(); in ExportBpropMindir() local227 if (prim_adapter == nullptr) { in ExportBpropMindir()231 auto prim = prim_adapter->attached_primitive(); in ExportBpropMindir()233 prim = std::make_shared<PrimitivePy>(obj, prim_adapter); in ExportBpropMindir()234 prim_adapter->set_attached_primitive(prim); in ExportBpropMindir()
254 auto prim_adapter = adapter_obj.cast<PrimitivePyAdapterPtr>(); in ConvertPrimitive() local255 MS_EXCEPTION_IF_NULL(prim_adapter); in ConvertPrimitive()256 auto primitive = prim_adapter->attached_primitive(); in ConvertPrimitive()258 primitive = std::make_shared<PrimitivePy>(adapter_obj, prim_adapter); in ConvertPrimitive()259 prim_adapter->set_attached_primitive(primitive); in ConvertPrimitive()
350 auto prim_adapter = obj_adapter.cast<PrimitivePyAdapterPtr>(); in Clone() local351 auto prim = std::make_shared<PrimitivePy>(obj_adapter, prim_adapter); in Clone()352 prim_adapter->set_attached_primitive(prim); in Clone()