Home
last modified time | relevance | path

Searched defs:ContextMapping (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dcreate_autodiff_subgraphs.cpp315 struct ContextMapping { struct
316 std::vector<const Node*> ctx_stack_;
317 std::unordered_map<const Node*, const Node*> node_to_ctx_;
319 void processNode(Node* n) { in processNode()
329 void processBlock(Block* block) { in processBlock()
342 ContextMapping(const std::shared_ptr<Graph>& graph) { in ContextMapping() argument
347 const Node* get(const Node* n) const { in get()
355 bool has(const Node* n) const { in has()