Searched defs:nodes_map (Results 1 – 8 of 8) sorted by relevance
311 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode()317 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode()382 const std::map<string, std::unique_ptr<TFGraphNode>>& nodes_map) { in ToProto()
20 std::unordered_map<const Node*, Node*> nodes_map; member
30 std::unordered_map<const Node*, Node*> nodes_map() const { in nodes_map() function in torch::jit::__anon32ccba180111::SubgraphMatcher
33 MemoryNodesMap&& nodes_map) in RawProcessMemoryNode()
40 nodes_map: Dict[Node, Node] = field(default_factory=dict) variable in InternalMatch
103 RawProcessMemoryNode::MemoryNodesMap nodes_map; in ReadProtoSnapshot() local
536 def nodes_map(nodes: List[torch.fx.Node], node_call_back) -> List[torch.fx.Node]: function
1081 std::map<string, int>* nodes_map, in VerifyConstants()