Searched refs:values_to_observe (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | insert_observers.cpp | 370 std::unordered_map<Value*, Module>& values_to_observe,
1363 std::unordered_map<Value*, Module>& values_to_observe, in recordObserved() argument
1369 values_to_observe[to_observe] = observer_module; in recordObserved()
1451 std::unordered_map<Value*, Module> values_to_observe; in insertObserversFor() local
1454 if (!inputs_outputs.count(v) && !values_to_observe.count(v)) { in insertObserversFor()
1457 v, *observer_opt, values_to_observe, block_observed_values); in insertObserversFor()
1511 values_to_observe, in insertObserversFor()
1521 values_to_observe, in insertObserversFor()
1549 values_to_observe, in insertObserversFor()
1595 v, *observer_opt, values_to_observe, block_observed_values); in insertObserversFor()
[all …]
|