Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dcreate_autodiff_subgraphs.cpp317 std::unordered_map<const Node*, const Node*> node_to_ctx_; member
320 node_to_ctx_[n] = ctx_stack_.back(); in processNode()
348 auto it = node_to_ctx_.find(n); in get()
350 it != node_to_ctx_.end(), in get()
356 return node_to_ctx_.find(n) != node_to_ctx_.end(); in has()