Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/ir/
Dprimal_attr.h38 …void SetPrimalAttr(const std::unordered_map<std::string, ValuePtr> &primal_attrs) { primal_attrs_ in SetPrimalAttr()
39 void ClearPrimalAttr() { primal_attrs_.clear(); } in ClearPrimalAttr()
40 std::unordered_map<std::string, ValuePtr> GetCurrentPrimalAttr() { return primal_attrs_; } in GetCurrentPrimalAttr()
44 std::unordered_map<std::string, ValuePtr> primal_attrs_; variable
Danf.h256 primal_attrs_ = PrimalAttrManager::GetInstance().GetCurrentPrimalAttr(); in CNode()
314 const std::unordered_map<std::string, ValuePtr> &primal_attrs() const { return primal_attrs_; } in primal_attrs()
316 primal_attrs_.insert(attrs.cbegin(), attrs.cend()); in set_primal_attrs()
318 void AddPrimalAttr(const std::string &name, const ValuePtr &attr) { primal_attrs_[name] = attr; } in AddPrimalAttr()
319 void ErasePrimalAttr(const std::string &name) { (void)primal_attrs_.erase(name); } in ErasePrimalAttr()
321 auto iter = primal_attrs_.find(name); in GetPrimalAttr()
322 return iter == primal_attrs_.cend() ? nullptr : iter->second; in GetPrimalAttr()
324 …bool HasPrimalAttr(const std::string &name) const { return primal_attrs_.find(name) != attrs_.cend… in HasPrimalAttr()
372 std::unordered_map<std::string, ValuePtr> primal_attrs_; variable
Danf.cc40 primal_attrs_ = PrimalAttrManager::GetInstance().GetCurrentPrimalAttr(); in CNode()