Searched defs:fake_quant_enabled (Results 1 – 4 of 4) sorted by relevance
60 const Tensor& fake_quant_enabled, in fake_quantize_tensor_cachemask_tensor_qparams_kernel_cuda()
96 const Tensor& fake_quant_enabled, in _fake_quantize_per_tensor_affine_cachemask_tensor_qparams()
132 auto fake_quant_enabled = at::ones(1, x.options().dtype(at::kLong)); in choose_qparams_fake_quant() local
2581 const Tensor& fake_quant_enabled,