Home
last modified time | relevance | path

Searched refs:node_attr_to_name_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfunction_extraction.cpp70 NodeAttrNameMap node_attr_to_name_; member
219 auto n_attr_it = node_attr_to_name_[n_in_def][attr.toUnqualString()] = name; in SetAttrName()
231 auto n_attr_it = node_attr_to_name_.find(n_in_def); in FindAttrName()
232 if (n_attr_it == node_attr_to_name_.end()) { in FindAttrName()
1063 auto func_ref_map = it.second->node_attr_to_name_; in run()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dexport.cpp411 NodeAttrNameMap node_attr_to_name_; member in torch::jit::__anon553abade0111::GraphEncoder
527 node_attr_to_name_(std::move(node_attr_to_name)) { in GraphEncoder()
973 auto attrs_it = node_attr_to_name_.find(node); in EncodeNode()
979 if (attrs_it != node_attr_to_name_.end()) { in EncodeNode()