Searched defs:inv_scale_ptr (Results 1 – 2 of 2) sorted by relevance
65 auto* inv_scale_ptr = inv_scale.data_ptr<float>(); in _amp_foreach_non_finite_check_and_unscale_cpu_kernel() local99 auto* inv_scale_ptr = inv_scale.data_ptr<float>(); in _amp_foreach_non_finite_check_and_unscale_cpu_kernel() local
60 auto* inv_scale_ptr = inv_scale.const_data_ptr<float>(); in _amp_non_finite_check_and_unscale_cuda_() local155 auto* inv_scale_ptr = inv_scale.const_data_ptr<float>(); in _amp_foreach_non_finite_check_and_unscale_cuda_() local