Home
last modified time | relevance | path

Searched refs:OpAdapterPtr (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter_desc.h29 …OpAdapterDesc(const OpAdapterPtr &train, const OpAdapterPtr &infer) : train_(train), infer_(infer)… in OpAdapterDesc()
31 explicit OpAdapterDesc(const OpAdapterPtr &common) : train_(common), infer_(common) {} in OpAdapterDesc()
47 OpAdapterPtr Get(bool train) const { return train ? train_ : infer_; } in Get()
68 OpAdapterPtr train_;
69 OpAdapterPtr infer_;
Dconvert.h75 static void RegisterAdapter(const std::string &name, OpAdapterPtr adpt);
76 …static void RegisterAdapter(const std::string &name, OpAdapterPtr train_adpt, OpAdapterPtr infer_a…
117 void SetOpInput(const OpAdapterPtr &adpt, const CNodePtr &node);
128 static OpAdapterPtr FindAdapter(const std::string &op_name, bool train = false);
129 static OpAdapterPtr FindAdapter(AnfNodePtr node, bool train = false);
175 void DrawCNode(const CNodePtr node, const OpAdapterPtr adpt);
188 …void SetTupleOpInput(const OpAdapterPtr &adpt, const CNodePtr &node, const AnfNodePtr &pred, const…
Dconvert.cc110 OpAdapterPtr DfGraphConvertor::FindAdapter(const AnfNodePtr node, bool train) { in FindAdapter()
133 return OpAdapterPtr(nullptr); in FindAdapter()
210 OpAdapterPtr DfGraphConvertor::FindAdapter(const std::string &name, bool train) { in FindAdapter()
762 OpAdapterPtr adpt = FindAdapter(node, training_); in SetSubgraph()
1218 void DfGraphConvertor::SetTupleOpInput(const OpAdapterPtr &adpt, const CNodePtr &node, const AnfNod… in SetTupleOpInput()
1294 void DfGraphConvertor::SetOpInput(const OpAdapterPtr &adpt, const CNodePtr &node) { in SetOpInput()
1360 OpAdapterPtr adpt = FindAdapter(cnode, training_); in SetNodeInput()
1400 OpAdapterPtr adpt = FindAdapter(node, training_); in UpdateOpDesc()
1481 OpAdapterPtr adpt = FindAdapter(value_ptr, training_); in ConvertTopK()
1526 OpAdapterPtr adpt = FindAdapter(node, training_); in ConvertReshape()
[all …]
Dop_adapter_base.h167 using OpAdapterPtr = std::shared_ptr<BaseOpAdapter>; variable
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute_ge.cc83 transform::OpAdapterPtr adapter = transform::DfGraphConvertor::FindAdapter(op_name, true); in SetInputsForSingleOpGraph()
126 transform::OpAdapterPtr adapter = transform::DfGraphConvertor::FindAdapter(op_name, true); in BuildSingleOpGraph()
272 transform::OpAdapterPtr adapter = transform::DfGraphConvertor::FindAdapter(op_name, true); in RunOpInGE()