Searched refs:primal_attrs (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | primal_attr.h | 38 …Attr(const std::unordered_map<std::string, ValuePtr> &primal_attrs) { primal_attrs_ = primal_attrs… in SetPrimalAttr() argument
51 explicit PrimalAttrGuard(const std::unordered_map<std::string, ValuePtr> &primal_attrs) { in PrimalAttrGuard() argument
52 PrimalAttrManager::GetInstance().SetPrimalAttr(primal_attrs); in PrimalAttrGuard()
|
| D | anf.h | 314 const std::unordered_map<std::string, ValuePtr> &primal_attrs() const { return primal_attrs_; } in primal_attrs() function
345 set_primal_attrs(node->primal_attrs()); in CloneCNodeInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | dfunctor.h | 170 … const CNodePtr &cnode, const std::unordered_map<std::string, ValuePtr> &primal_attrs,
188 … const CNodePtr &cnode, const std::unordered_map<std::string, ValuePtr> &primal_attrs, in BpropToK() argument
196 PrimalAttrGuard primal_attr_guard(primal_attrs); in BpropToK()
|
| D | kprim.cc | 439 std::unordered_map<std::string, ValuePtr> primal_attrs; in KPrimitive() local
449 primal_attrs = cnode->primal_attrs(); in KPrimitive()
451 primal_attrs[kPrimalAttrForwardNodeName] = MakeValue(forward_node_primal_attr); in KPrimitive()
453 auto expanded_fg = BpropToK(prim, bprop_fg, nullptr, cnode, primal_attrs, primal_debug_infos); in KPrimitive()
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | anf_ir_dump.cc | 374 if (op->primal_attrs().empty()) { in DumpCNodePrimalAttrs()
378 auto primal_attrs = op->primal_attrs(); in DumpCNodePrimalAttrs() local
380 DumpAttrs(primal_attrs, gsub); in DumpCNodePrimalAttrs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | shape_ops_splitter.cc | 48 node->set_primal_attrs(cnode->primal_attrs()); in CloneCNode()
|
| D | graph_kernel_splitter.cc | 95 cp_node->set_primal_attrs(cnode->primal_attrs()); in NewRecomputeNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 760 replace_node->set_primal_attrs(node->primal_attrs()); in StepReplaceOp()
766 new_cnode->set_primal_attrs(node->primal_attrs()); in StepReplaceOp()
808 replace_output->set_primal_attrs(node->primal_attrs()); in StepReplaceGraph()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.cc | 435 cnode->set_primal_attrs(ori_cnode->primal_attrs()); in NewCNodeWithInfos()
|