Searched defs:fake_quant_on (Results 1 – 5 of 5) sorted by relevance
26 const int64_t* fake_quant_on, in ChooseQuantizationParamsKernelImpl()186 const at::Tensor fake_quant_on, in _calc_moving_avg_qparams_helper()239 const at::Tensor& fake_quant_on, in fused_moving_avg_obs_fake_quant_cuda()
65 int64_t* fake_quant_on = fake_quant_enabled.data_ptr<int64_t>(); in fake_quantize_tensor_cachemask_tensor_qparams_kernel_cuda() local
147 const at::Tensor& fake_quant_on, in fused_moving_avg_obs_fake_quant_cpu()225 const at::Tensor& fake_quant_on, in fused_moving_avg_obs_fake_quant()
2073 fake_quant_on, argument
2529 int fake_quant_on,