Searched defs:found_inf_ptr (Results 1 – 11 of 11) sorted by relevance
64 auto* found_inf_ptr = found_inf.data_ptr<float>(); in _amp_foreach_non_finite_check_and_unscale_cpu_kernel() local98 auto* found_inf_ptr = found_inf.data_ptr<float>(); in _amp_foreach_non_finite_check_and_unscale_cpu_kernel() local169 float* found_inf_ptr = found_inf.data_ptr<float>(); in _amp_update_scale_cpu_kernel() local
163 float* found_inf_ptr = in _fused_sgd_with_momentum_kernel_cuda_() local238 float* found_inf_ptr = in _fused_sgd_with_momentum_kernel_cuda_() local304 float* found_inf_ptr = in _fused_sgd_kernel_cuda_() local397 float* found_inf_ptr = in _fused_sgd_kernel_cuda_() local
31 const float* found_inf_ptr = in _fused_adamw_cuda_impl_() local77 const float* found_inf_ptr = in _fused_adamw_cuda_impl_() local
30 const float* found_inf_ptr = in _fused_adam_cuda_impl_() local76 const float* found_inf_ptr = in _fused_adam_cuda_impl_() local
35 const float* found_inf_ptr = in _fused_adam_amsgrad_cuda_impl_() local86 const float* found_inf_ptr = in _fused_adam_amsgrad_cuda_impl_() local
36 const float* found_inf_ptr = in _fused_adamw_amsgrad_cuda_impl_() local87 const float* found_inf_ptr = in _fused_adamw_amsgrad_cuda_impl_() local
59 auto* found_inf_ptr = found_inf.mutable_data_ptr<float>(); in _amp_non_finite_check_and_unscale_cuda_() local154 auto* found_inf_ptr = found_inf.mutable_data_ptr<float>(); in _amp_foreach_non_finite_check_and_unscale_cuda_() local
36 const float* found_inf_ptr, in adam_math()
31 const float* found_inf_ptr = in _fused_adagrad_kernel_cpu_() local
37 const float* found_inf_ptr = in _fused_adam_kernel_cpu_() local113 const float* found_inf_ptr = in _fused_adamw_kernel_cpu_() local
33 const float* found_inf_ptr = in _fused_sgd_kernel_cpu_() local