Searched defs:grad_factor (Results 1 – 4 of 4) sorted by relevance
99 float grad_factor) { in _fake_quantize_grad_learnable_tensor_kernel_cuda()190 …hannel_kernel_cuda(TensorIterator &iter, int64_t quant_min, int64_t quant_max, float grad_factor) { in _fake_quantize_grad_learnable_channel_kernel_cuda()
148 double grad_factor) { in _fake_quantize_learnable_per_channel_affine()162 double grad_factor) { in _fake_quantize_learnable_per_channel_affine_backward()
154 double grad_factor) { in _fake_quantize_learnable_per_tensor_affine()168 double grad_factor) { in _fake_quantize_learnable_per_tensor_affine_backward()
2594 float grad_factor) {2719 float grad_factor) {