Searched defs:neginf (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseUnaryOps.cpp | 263 std::optional<double> posinf, std::optional<double> neginf) { in nan_to_num_sparse()
271 std::optional<double> posinf, std::optional<double> neginf, in nan_to_num_sparse_out()
280 std::optional<double> posinf, std::optional<double> neginf) { in nan_to_num_sparse_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | LossCTC.cu | 87 constexpr scalar_t neginf = -INFINITY; in ctc_loss_log_alpha_gpu_kernel() local
329 constexpr scalar_t neginf = -INFINITY; in ctc_loss_backward_log_beta_gpu_kernel() local
517 constexpr scalar_t neginf = -INFINITY; in ctc_loss_backward_collect_gpu_kernel() local
607 constexpr scalar_t neginf = -INFINITY; in ctc_loss_backward_gpu_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UnaryOpsKernel.cpp | 465 c10::complex<scalar_t> a, scalar_t nan, scalar_t posinf, scalar_t neginf) { in _nan_to_num_replace()
474 Vectorized<scalar_t> a, scalar_t nan, scalar_t posinf, scalar_t neginf) { in _nan_to_num_replace()
485 Vectorized<c10::complex<scalar_t>> a, scalar_t nan, scalar_t posinf, scalar_t neginf) { in _nan_to_num_replace()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossCTC.cpp | 132 constexpr scalar_t neginf = -std::numeric_limits<scalar_t>::infinity(); in ctc_loss_cpu_template() local
235 constexpr scalar_t neginf = -std::numeric_limits<scalar_t>::infinity(); in ctc_loss_backward_cpu_template() local
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset10.py | 752 def nan_to_num(g: jit_utils.GraphContext, input, nan, posinf, neginf): argument
|
| /aosp_15_r20/external/starlark-go/starlark/ |
| H A D | library.go | 417 neginf = Float(math.Inf(-1)) var
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _funcs_impl.py | 859 x: ArrayLike, copy: NotImplementedType = True, nan=0.0, posinf=None, neginf=None argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6270 def nan_to_num(self, nan=None, posinf=None, neginf=None): argument
|