Home
last modified time | relevance | path

Searched refs:PrimitivePyAdapter (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/pybind_api/ir/
Dprimitive_py.cc392 PrimitivePyAdapter::PrimitivePyAdapter(const py::str &name) : name_(name) {} in PrimitivePyAdapter() function in mindspore::PrimitivePyAdapter
394 void PrimitivePyAdapter::AddPyAttr(const py::str &name, const py::object &obj) { in AddPyAttr()
429 void PrimitivePyAdapter::DelPyAttr(const py::str &name) { in DelPyAttr()
437 py::dict PrimitivePyAdapter::GetAttrDict() { in GetAttrDict()
450 void PrimitivePyAdapter::set_prim_type(const PrimType t) { in set_prim_type()
457 void PrimitivePyAdapter::set_const_prim(bool is_const_prim) { in set_const_prim()
464 void PrimitivePyAdapter::set_const_input_indexes(const std::vector<size_t> &const_input_indexes) { in set_const_input_indexes()
472 void PrimitivePyAdapter::set_signatures(const std::vector<Signature> &signatures) { in set_signatures()
480 void PrimitivePyAdapter::set_hook(const py::function &hook) { in set_hook()
488 void PrimitivePyAdapter::set_instance_name(const std::string &s) { in set_instance_name()
[all …]
Dprimitive_py.h42 class PrimitivePyAdapter; variable
43 using PrimitivePyAdapterPtr = std::shared_ptr<PrimitivePyAdapter>;
89 class PrimitivePyAdapter {
91 explicit PrimitivePyAdapter(const py::str &name);
92 ~PrimitivePyAdapter() = default;
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpattern.h93 if (py::isinstance<PrimitivePyAdapter>(prim_obj)) { in Prim()
132 if (py::isinstance<PrimitivePyAdapter>(prim_obj)) { in Call()