Home
last modified time | relevance | path

Searched refs:configure_constructor_to_put_obs_on_module_device (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/ao/quantization/
Dqconfig.py596 def configure_constructor_to_put_obs_on_module_device(original_constructor): function
609 activation = configure_constructor_to_put_obs_on_module_device(qconfig.activation)
610 weight = configure_constructor_to_put_obs_on_module_device(qconfig.weight)