Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/runtime/interpreter/
Dpreprocess_graph.cpp100 struct InsertLastUses { in insertLastUses() struct
101 Graph& graph; in insertLastUses()
103 std::unordered_set<Value*> seen; in insertLastUses()
108 std::unordered_map<Node*, Node*> drop_for_node; in insertLastUses()
110 explicit InsertLastUses(Graph& g) : graph(g) { in insertLastUses() argument
113 void scanBlock(Block* b) { in insertLastUses()
119 void scanNode(Node* n) { in insertLastUses()
129 void scanUse(Node* n, size_t i) { in insertLastUses()
171 Node* findOwnerInBlock(Node* n, Block* block) { in insertLastUses()
178 Node* findOrCreateDropInstructionForNode(Node* n) { in insertLastUses()
[all …]