Searched refs:externalValuesMap (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/csrc/jit/passes/utils/ |
| D | subgraph_utils.cpp | 233 std::unordered_map<Value*, Value*>& externalValuesMap, in collectNestedUses() argument
236 if (externalValuesMap.count(input) == 0 && new_values.count(input) == 0) { in collectNestedUses()
244 closed_over_values, new_values, externalValuesMap, node); in collectNestedUses()
247 if (externalValuesMap.count(v) == 0 && new_values.count(v) == 0) { in collectNestedUses()
254 if (externalValuesMap.count(v) == 0 && new_values.count(v) == 0) { in collectNestedUses()
264 closed_over_values, new_values, externalValuesMap, node); in collectNestedUses()
276 std::unordered_map<Value*, Value*>& externalValuesMap) { in closedOverValues() argument
279 collectNestedUses(closed_over_values, new_values, externalValuesMap, toMerge); in closedOverValues()
295 std::unordered_map<Value*, Value*> externalValuesMap; in mergeNodeIntoSubgraph() local
300 externalValuesMap[input] = subgraph->inputs()[idx]; in mergeNodeIntoSubgraph()
[all …]
|