Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/quantization/
Dinsert_observers.cpp949 const Module& observer_module, in insertObserverFor()
1238 Module observer_module = getObserverModuleFor(v, qconfig); in valueNeedsToBeQuantized() local
1362 const Module& observer_module, in recordObserved()
Dinsert_quant_dequant.cpp254 auto observer_module = module.attr(observer_name.value()).toModule(); in getObserverDtype() local
266 auto observer_module = module.attr(findObserverName(v).value()).toModule(); in getEmbeddingBagObsName() local
1069 auto observer_module = module.attr(observer_name.value()).toModule(); in getQSchemeAndQParamVector() local
1566 auto observer_module = module.attr(observer_name.value()).toModule(); in insertCalculateQParams() local