Searched defs:nan_replacement (Results 1 – 2 of 2) sorted by relevance
212 C10_HOST_DEVICE static inline scalar_t _nan_to_num_replace(scalar_t a, scalar_t nan_replacement, sc… in _nan_to_num_replace()230 value_t nan_replacement = static_cast<value_t>(nan.value_or(0.)); in nan_to_num_kernel_cuda() local248 scalar_t nan_replacement = static_cast<scalar_t>(nan.value_or(0.)); in nan_to_num_kernel_cuda() local
451 …scalar_t a, scalar_t nan_replacement, scalar_t pos_inf_replacement, scalar_t neg_inf_replacement) { in _nan_to_num_replace()504 value_t nan_replacement = static_cast<value_t>(nan.value_or(0.)); in nan_to_num_kernel() local