Searched defs:pos_inf_replacement (Results 1 – 2 of 2) sorted by relevance
212 …r_t _nan_to_num_replace(scalar_t a, scalar_t nan_replacement, scalar_t pos_inf_replacement, scalar… in _nan_to_num_replace()231 value_t pos_inf_replacement = pos_inf.has_value() in nan_to_num_kernel_cuda() local249 scalar_t pos_inf_replacement = pos_inf.has_value() 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()505 value_t pos_inf_replacement = pos_inf.has_value() in nan_to_num_kernel() local