Home
last modified time | relevance | path

Searched refs:OperatorPtr (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter_base.h69 OperatorPtr op;
73 OutHandler(const OperatorPtr &op, const std::string out, const AnfNodePtr &node = nullptr)
78 OperatorPtr src_op;
79 OperatorPtr dest_op;
82 using AttrFunc = std::function<void(OperatorPtr, ValuePtr)>;
83 using OutputFunc = std::function<OutHandler(OperatorPtr)>;
84 using InputOpFunc = std::function<void(OperatorPtr, OperatorPtr)>;
85 using InputHandleFunc = std::function<void(OperatorPtr, OutHandler)>;
86 using CreateDynInputOpFunc = std::function<void(OperatorPtr, unsigned int)>;
87 using DynInputOpFunc = std::function<void(OperatorPtr, unsigned int, OperatorPtr)>;
[all …]
Dop_adapter.h63 bool IsCustomOp(const OperatorPtr &op);
66 OperatorPtr GenerateCustomOp(const AnfNodePtr anf);
67 …Status SetOpSubgraphFunc(const OperatorPtr &op, int index, const std::shared_ptr<std::vector<DfGra…
68 Status SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &input);
69 Status SetNormalOpInput(const OperatorPtr &op, int index, const OperatorPtr &input);
70 int setInput(const OperatorPtr &op, int index, const OperatorPtr &input);
72 Status SetNormalOpInput(const OperatorPtr &op, int index, const OutHandler &handle);
73 int setInput(const OperatorPtr &op, int index, const OutHandler &handle);
74 …int setInput(const OperatorPtr &op, int index, const std::shared_ptr<std::vector<OutHandler>> &han…
75 OutHandler getOutput(const OperatorPtr &op, int index);
[all …]
Dconvert.h151 OperatorPtr Convert(AnfNodePtr node);
152 OperatorPtr ConvertCNode(CNodePtr node);
153 std::vector<OperatorPtr> ConvertDependNode(AnfNodePtr node);
155 OperatorPtr ConvertParameter(AnfNodePtr node);
157 OperatorPtr ConvertValueNode(ValueNodePtr node);
176 void UpdateDataOpDesc(const AnfNodePtr &it, const OperatorPtr &op) const;
177 void AddGraphConstInput(const OperatorPtr &op);
178 OperatorPtr ToOperatorPtr(const AnfNodePtr &node);
188 …put(const OpAdapterPtr &adpt, const CNodePtr &node, const AnfNodePtr &pred, const OperatorPtr &src,
200 std::unordered_map<AnfNode *, OperatorPtr> op_cache_;
[all …]
Dop_adapter.cc23 bool OpAdapterImpl::IsCustomOp(const OperatorPtr &op) { in IsCustomOp()
81 OperatorPtr OpAdapterImpl::GenerateCustomOp(const AnfNodePtr anf) { in GenerateCustomOp()
108 Status OpAdapterImpl::SetOpSubgraphFunc(const OperatorPtr &op, int index, in SetOpSubgraphFunc()
123 Status OpAdapterImpl::SetCustomOpInput(const CusOperatorPtr &op, int index, const OperatorPtr &inpu… in SetCustomOpInput()
140 Status OpAdapterImpl::SetNormalOpInput(const OperatorPtr &op, int index, const OperatorPtr &input) { in SetNormalOpInput()
151 int OpAdapterImpl::setInput(const OperatorPtr &op, int index, const OperatorPtr &input) { in setInput()
182 Status OpAdapterImpl::SetNormalOpInput(const OperatorPtr &op, int index, const OutHandler &handle) { in SetNormalOpInput()
199 int OpAdapterImpl::setInput(const OperatorPtr &op, int index, const OutHandler &handle) { in setInput()
208 int OpAdapterImpl::setInput(const OperatorPtr &op, int index, in setInput()
236 OutHandler OpAdapterImpl::getOutput(const OperatorPtr &op, int index) { in getOutput()
[all …]
Dconvert.cc434 …ize = std::count_if(vars_.begin(), vars_.end(), [](const std::pair<std::string, OperatorPtr> &it) { in BuildSaveCheckpointGraph()
708 void SetupDatasetIterGetNextNode(const OperatorPtr &op) { in SetupDatasetIterGetNextNode()
768 OperatorPtr op = Convert(node); in SetSubgraph()
926 [](OperatorPtr x) { return *x; }); in BuildGraph()
945 void DfGraphConvertor::UpdateDataOpDesc(const AnfNodePtr &it, const OperatorPtr &op) const { in UpdateDataOpDesc()
1060 OperatorPtr DfGraphConvertor::ToOperatorPtr(const AnfNodePtr &node) { in ToOperatorPtr()
1219 const OperatorPtr &src, int index) { in SetTupleOpInput()
1295 OperatorPtr src = Convert(node); in SetOpInput()
1346 void DfGraphConvertor::AddGraphConstInput(const OperatorPtr &op) { in AddGraphConstInput()
1407 OperatorPtr op = Convert(node); in UpdateOpDesc()
[all …]
Dtypes.h52 using OperatorPtr = std::shared_ptr<ge::Operator>; variable
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dop_declare_macro.h66 [](const OperatorPtr op, const OperatorPtr input) { \
70 [](const OperatorPtr op, const OutHandler& handle) { \
74 [](const OperatorPtr op, const GeTensorDesc desc) { \
86 [](const OperatorPtr op, unsigned int num) { \
90 [](const OperatorPtr op, unsigned int index, const OperatorPtr input) { \
94 [](const OperatorPtr op, unsigned int index, const OutHandler& handle) { \
106 [](const OperatorPtr op, unsigned int num) { \
110 [](const OperatorPtr op, unsigned int index, const DfGraphPtr graph) { \
123 [](const OperatorPtr op, const ValuePtr& value) { \
139 [](const OperatorPtr op, const GeTensorDesc desc) { \
[all …]
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute_ge.cc46 using transform::OperatorPtr;
78 const OperatorPtr &op, std::vector<GeOperator> *graph_input_nodes) { in SetInputsForSingleOpGraph()
131 OperatorPtr op = adapter->generate(op_name); in BuildSingleOpGraph()