Searched refs:nodes_map_ (Results 1 – 4 of 4) sorted by relevance
100 explicit CNodeDecoder(std::map<std::string, AnfNodePtr> *nodes_map) : nodes_map_(*nodes_map) {} in CNodeDecoder()183 } else if (nodes_map_.count(name) == 0) { in DecodeInputDesc()187 inputs.push_back(nodes_map_[name]); in DecodeInputDesc()210 nodes_map_[output_desc[kJsonKeyTensorName]] = cnode_; in DecodeOutputDesc()221 nodes_map_[output_desc[kJsonKeyTensorName]] = get_item; in DecodeOutputDesc()320 std::map<std::string, AnfNodePtr> &nodes_map_; member in mindspore::kernel::__anonc55854f90111::CNodeDecoder353 nodes_map_[name] = new_parameter; in DecodeParameter()359 CNodeDecoder decoder(&nodes_map_); in DecodeCNode()370 if (nodes_map_.count(name) == 0) { in DecodeOutput()374 outputs.push_back(nodes_map_[name]); in DecodeOutput()[all …]
31 AkgKernelJsonDecoder() { nodes_map_.clear(); } in AkgKernelJsonDecoder()41 std::map<std::string, AnfNodePtr> nodes_map_; variable
418 auto &nodes = nodes_map_[key]; in InitSomasStreamAndNode()430 auto nodes = nodes_map_[kernel.get()]; in InitSomasOutputAndWorkspaceTensors()507 auto nodes = nodes_map_[kernel.get()]; in InitCommonNodeInputs()542 auto iter = nodes_map_.find(prenode_index.first.get()); in InitCommonNodeInputs()543 if (iter == nodes_map_.end()) { in InitCommonNodeInputs()581 auto node = nodes_map_[kernel.get()].at(0); in InitAtomicCleanInputs()590 auto iter = nodes_map_.find(pre_node.get()); in InitAtomicCleanInputs()591 if (iter == nodes_map_.end()) { in InitAtomicCleanInputs()707 auto iter = nodes_map_.find(kernel.get()); in GetNextOutputProcess()708 if (iter != nodes_map_.end()) { in GetNextOutputProcess()[all …]
68 std::map<void *, std::vector<SomasNodePtr>> nodes_map_; variable