| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | nan_to_num_test.py | 8 """Microbenchmarks for torch.nan_to_num / nan_to_num_ operators"""
10 # Configs for PT torch.nan_to_num / nan_to_num_ operators
16 ["nan_to_num_", torch.nan_to_num_],
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops_util.cpp | 48 tensor1.nan_to_num_(); in EqualValues()
49 tensor2.nan_to_num_(); in EqualValues()
103 tensor1.nan_to_num_(); in CloseValues()
104 tensor2.nan_to_num_(); in CloseValues()
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 424 - aten::nan_to_num_
903 aten::nan_to_num_: 10
|
| H A D | model_ops.yaml | 253 aten::nan_to_num_: 13
|
| H A D | math_ops.py | 163 torch.nan_to_num_(w),
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseUnaryOps.cpp | 281 TORCH_CHECK(self.is_coalesced(), "nan_to_num_ requires coalesced input"); in nan_to_num_sparse_()
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | __init__.py | 371 aten.nan_to_num_,
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _pointwise_ops.py | 311 aten.nan_to_num_.default,
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | tensors.rst | 601 Tensor.nan_to_num_
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | passes.cpp | 56 %y3 = aten::nan_to_num_(%y2, %f, %g, %h) in ConcatAddMulReplaceNaNClip()
64 %y3 = aten::nan_to_num_(%y2, %f, %g, %h) in ConcatAddMulReplaceNaNClip()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UnaryOps.cpp | 835 Tensor& nan_to_num_( in nan_to_num_() function
|
| H A D | LinearAlgebra.cpp | 3155 result.nan_to_num_(INFINITY, INFINITY, -INFINITY); in _linalg_cond_helper()
|
| H A D | native_functions.yaml | 3282 - func: nan_to_num_(Tensor(a!) self, float? nan=None, float? posinf=None, float? neginf=None) -> Te…
3285 CompositeExplicitAutograd: nan_to_num_
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebraLib.cpp | 1497 at::nan_to_num_(const_cast<Tensor&>(self), 0, std::numeric_limits<double>::infinity(), in lu_factor_looped_cusolver()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _tensor_docs.py | 3508 "nan_to_num_",
3510 nan_to_num_(nan=0.0, posinf=None, neginf=None) -> Tensor
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1990 "torch.nan_to_num_",
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nestedtensor.py | 1801 expect = torch.nn.functional.softmax(pt, 1).nan_to_num_(0.0)
3489 torch.nan_to_num_(deviation, nan=default_rtol[computed_value.dtype])
|
| H A D | test_transformers.py | 83 torch.nan_to_num_(deviation, nan=default_rtol[computed_value.dtype])
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 6299 nan_to_num_ = _make_inplace(nan_to_num) variable
|