Home
last modified time | relevance | path

Searched refs:_numpy (Results 1 – 25 of 50) sorted by relevance

12

/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/fft/
H A Dtest_helper.py20 import torch._numpy as np
21 from torch._numpy import fft, pi
22 from torch._numpy.testing import assert_array_almost_equal
90 from torch._numpy import arange, asarray, concatenate, take
H A Dtest_pocketfft.py25 import torch._numpy as np
26 from torch._numpy.random import random
27 from torch._numpy.testing import assert_allclose # , IS_WASM
/aosp_15_r20/external/pytorch/torch/_numpy/
H A DREADME.md16 accept. In particular, they accept `torch._numpy.dtype`s or `torch._numpy.ndarray`s.
42 simply by changing `import numpy as np` by `import torch._numpy as np` in your program, without hav…
43 Note that `torch._numpy` will be quite slow when used in eager mode, and it is in no way a replace…
/aosp_15_r20/external/pytorch/test/torch_np/
H A Dtest_nep50_examples.py17 import torch._numpy as tnp
18 from torch._numpy import ( # noqa: F401
31 from torch._numpy.testing import assert_allclose
H A Dtest_function_base.py19 import torch._numpy as np
20 from torch._numpy.testing import assert_equal
H A Dtest_scalars_0D_arrays.py27 import torch._numpy as np
28 from torch._numpy.testing import assert_equal
H A Dtest_basic.py11 import torch._numpy as w
12 import torch._numpy._ufuncs as _ufuncs
13 import torch._numpy._util as _util
14 from torch._numpy.testing import assert_allclose, assert_equal
H A Dtest_random.py12 import torch._numpy as tnp
13 from torch._numpy.testing import assert_equal
H A Dtest_binary_ufuncs.py7 from torch._numpy._ufuncs import * # noqa: F403
8 from torch._numpy.testing import assert_allclose
H A Dtest_unary_ufuncs.py8 from torch._numpy._ufuncs import * # noqa: F403
9 from torch._numpy.testing import assert_allclose
H A Dtest_reductions.py31 import torch._numpy as np
32 from torch._numpy import _util
33 from torch._numpy.testing import (
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/
H A Dtest_getlimits.py29 import torch._numpy as np
30 from torch._numpy import double, finfo, half, iinfo, single
31 from torch._numpy.testing import assert_, assert_equal
H A Dtest_scalarinherit.py11 import torch._numpy as np
12 from torch._numpy.testing import assert_
H A Dtest_scalar_ctors.py26 import torch._numpy as np
27 from torch._numpy.testing import assert_almost_equal, assert_equal
H A Dtest_dlpack.py27 import torch._numpy as np
28 from torch._numpy.testing import assert_array_equal
H A Dtest_numerictypes.py25 import torch._numpy as np
26 from torch._numpy.testing import assert_
H A Dtest_scalar_methods.py29 import torch._numpy as np
30 from torch._numpy.testing import assert_equal
H A Dtest_dtype.py33 import torch._numpy as np
34 from torch._numpy.testing import assert_, assert_equal
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/
H A Dtest_arraypad.py19 import torch._numpy as np
20 from torch._numpy.testing import assert_allclose, assert_array_equal
H A Dtest_type_check.py32 import torch._numpy as np
33 from torch._numpy import (
44 from torch._numpy.testing import assert_, assert_array_equal, assert_equal
H A Dtest_twodim_base.py50 import torch._numpy as np
51 from torch._numpy import (
68 from torch._numpy.testing import (
H A Dtest_index_tricks.py35 import torch._numpy as np
36 from torch._numpy import (
43 from torch._numpy.testing import (
H A Dtest_shape_base_.py43 import torch._numpy as np
44 from torch._numpy import (
58 from torch._numpy.random import rand, randint
59 from torch._numpy.testing import assert_, assert_array_equal, assert_equal
/aosp_15_r20/external/pytorch/torch/_numpy/testing/
H A Dutils.py22 import torch._numpy as np
23 from torch._numpy import arange, asarray as asanyarray, empty, float32, intp, ndarray
220 from torch._numpy import imag, iscomplexobj, isscalar, ndarray, real, signbit
399 from torch._numpy import imag, iscomplexobj, ndarray, real
568 from torch._numpy import all, array, asarray, bool_, inf, isnan, max
917 from torch._numpy import any as npany, float_, issubdtype, number, result_type
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dops.py229 text = repr(tensor._numpy()) if is_repr else str(tensor._numpy())
1051 return complex(self._numpy())
1054 return int(self._numpy())
1057 return long(self._numpy())
1060 return float(self._numpy())
1063 return self._numpy().__index__()
1066 return bool(self._numpy())
1075 return self._numpy().__format__(format_spec)
1080 return convert_to_tensor, (self._numpy(),)
1110 a = self._numpy()
[all …]

12