Home
last modified time | relevance | path

Searched defs:nan_to_num (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset10.py752 def nan_to_num(g: jit_utils.GraphContext, input, nan, posinf, neginf): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUnaryOps.cpp826 Tensor nan_to_num( in nan_to_num() function
/aosp_15_r20/external/pytorch/torch/_numpy/
H A D_funcs_impl.py858 def nan_to_num( function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py816 def nan_to_num( function
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py6270 def nan_to_num(self, nan=None, posinf=None, neginf=None): function