Home
last modified time | relevance | path

Searched defs:observer_on (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/quantized/cuda/
DFusedObsFakeQuant.cu104 const int64_t* observer_on, in MovingAverageMinMax()
134 const at::Tensor& observer_on, in _calculate_moving_average()
238 const at::Tensor& observer_on, in fused_moving_avg_obs_fake_quant_cuda()
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dfused_obs_fake_quant.cpp146 const at::Tensor& observer_on, in fused_moving_avg_obs_fake_quant_cpu()
224 const at::Tensor& observer_on, in fused_moving_avg_obs_fake_quant()
/external/pytorch/torch/
D_meta_registrations.py2072 observer_on, argument