Searched refs:FuncGraphTransform (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph.h | 103 class FuncGraphTransform {
107 explicit FuncGraphTransform(const PrimitivePtr prim, const FuncGraphPtr func_graph = nullptr)
110 …explicit FuncGraphTransform(const FuncGraphPtr &func_graph, const PrimitivePtr &prim = func_graph_…
113 FuncGraphTransform(const FuncGraphTransform &t) : prim_(t.prim_), func_graph_(t.func_graph_) {} in FuncGraphTransform() function
115 ~FuncGraphTransform() = default;
130 FuncGraphTransform &operator=(const FuncGraphTransform &t) {
248 std::unordered_map<std::string, FuncGraphTransform> &transforms() { return transforms_; } in transforms()
249 void set_transforms(const std::unordered_map<std::string, FuncGraphTransform> &transforms) { in set_transforms()
366 std::unordered_map<std::string, FuncGraphTransform> transforms_;
|
| D | func_graph.cc | 773 const PrimitivePtr FuncGraphTransform::func_graph_prim_ = std::make_shared<Primitive>("FuncGraph");
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | grad.cc | 117 (void)func_graph->transforms().insert(std::make_pair("grad", FuncGraphTransform(res))); in Grad()
|
| D | dfunctor.cc | 474 (void)primal_graph_->transforms().insert(std::make_pair("grad", FuncGraphTransform(k_graph_))); in MapMorphism()
475 (void)k_graph_->transforms().insert(std::make_pair("primal", FuncGraphTransform(primal_graph_))); in MapMorphism()
500 (void)primal->transforms().insert(std::make_pair("grad", FuncGraphTransform(fg))); in KUserDefined()
501 (void)fg->transforms().insert(std::make_pair("primal", FuncGraphTransform(primal))); in KUserDefined()
|
| D | dfunctor.h | 229 (void)outer->transforms().emplace("primal", FuncGraphTransform(primal)); in BpropToK()
|
| D | kprim.cc | 429 fprop->transforms().emplace("primal", FuncGraphTransform(prim::kPrimSwitchLayer)); in KPrimitive()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | data_converter.cc | 326 …(void)func_graph->transforms().insert(std::make_pair(CUSTOM_BPROP_NAME, FuncGraphTransform(bprop_g… in ConvertCellObjToFuncGraph()
327 … (void)bprop_graph->transforms().insert(std::make_pair("primal", FuncGraphTransform(func_graph))); in ConvertCellObjToFuncGraph()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 517 (void)fg->transforms().emplace("primal", FuncGraphTransform(prim::kPrimMakeTuple)); in GenerateFuncGraph()
558 (void)fg->transforms().emplace("primal", FuncGraphTransform(prim::kPrimMakeList)); in GenerateFuncGraph()
|