quantization_pertensor_hook (reference) in projects: aosp_15_r20

Searched refs:quantization_pertensor_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	`62 _ddp_comm_hook_wrapper, comm_hook=quantization.quantization_pertensor_hook`
H A D	quantization_hooks.py	`46 def quantization_pertensor_hook( function`
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A D	distributed_test.py	`4498 quantization_hooks.quantization_pertensor_hook,`