quantization_perchannel_hook (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/
H A D	__init__.py	`65 _ddp_comm_hook_wrapper, comm_hook=quantization.quantization_perchannel_hook`
H A D	quantization_hooks.py	`121 def quantization_perchannel_hook( function`
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A D	distributed_test.py	`4499 quantization_hooks.quantization_perchannel_hook,`