| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _funcs_impl.py | 35 a: ArrayLike, order: NotImplementedType = "K", subok: NotImplementedType = False
42 src: ArrayLike,
50 def atleast_1d(*arys: ArrayLike):
58 def atleast_2d(*arys: ArrayLike):
66 def atleast_3d(*arys: ArrayLike):
112 ar_tuple: Sequence[ArrayLike],
124 tup: Sequence[ArrayLike],
138 tup: Sequence[ArrayLike],
149 tup: Sequence[ArrayLike],
162 tup: Sequence[ArrayLike],
[all …]
|
| H A D | linalg.py | 51 def matrix_power(a: ArrayLike, n):
58 def multi_dot(inputs: Sequence[ArrayLike], *, out=None):
67 def solve(a: ArrayLike, b: ArrayLike):
74 def lstsq(a: ArrayLike, b: ArrayLike, rcond=None):
84 def inv(a: ArrayLike):
92 def pinv(a: ArrayLike, rcond=1e-15, hermitian=False):
99 def tensorsolve(a: ArrayLike, b: ArrayLike, axes=None):
106 def tensorinv(a: ArrayLike, ind=2):
116 def det(a: ArrayLike):
123 def slogdet(a: ArrayLike):
[all …]
|
| H A D | _reductions_impl.py | 67 def count_nonzero(a: ArrayLike, axis: AxisLike = None, *, keepdims: KeepDims = False):
73 a: ArrayLike,
93 a: ArrayLike,
113 a: ArrayLike,
127 a: ArrayLike,
141 a: ArrayLike,
159 a: ArrayLike,
177 a: ArrayLike,
187 a: ArrayLike,
206 a: ArrayLike,
[all …]
|
| H A D | fft.py | 31 def fft(a: ArrayLike, n=None, axis=-1, norm=None):
37 def ifft(a: ArrayLike, n=None, axis=-1, norm=None):
43 def rfft(a: ArrayLike, n=None, axis=-1, norm=None):
49 def irfft(a: ArrayLike, n=None, axis=-1, norm=None):
55 def fftn(a: ArrayLike, s=None, axes=None, norm=None):
61 def ifftn(a: ArrayLike, s=None, axes=None, norm=None):
67 def rfftn(a: ArrayLike, s=None, axes=None, norm=None):
73 def irfftn(a: ArrayLike, s=None, axes=None, norm=None):
79 def fft2(a: ArrayLike, s=None, axes=(-2, -1), norm=None):
85 def ifft2(a: ArrayLike, s=None, axes=(-2, -1), norm=None):
[all …]
|
| H A D | _ufuncs.py | 123 x1: ArrayLike,
124 x2: ArrayLike,
190 x1: ArrayLike,
191 x2: ArrayLike,
298 x: ArrayLike,
|
| H A D | random.py | 157 def choice(a: ArrayLike, size=None, replace=True, p: Optional[ArrayLike] = None):
|
| H A D | _ndarray.py | 313 def copy(self: ArrayLike, order: NotImplementedType = "C"):
317 def flatten(self: ArrayLike, order: NotImplementedType = "C"):
358 def fill(self, value: ArrayLike):
|
| H A D | _normalizations.py | 17 ArrayLike = typing.TypeVar("ArrayLike") variable
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | device_mesh.py | 425 mesh: Union[torch.Tensor, "ArrayLike"],
737 mesh: Optional[Union[torch.Tensor, "ArrayLike"]] = None,
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
| H A D | test_scalarmath.py | 641 class ArrayLike: class
|
| H A D | test_multiarray.py | 6092 class ArrayLike: class
6110 class ArrayLike: class
|
| /aosp_15_r20/external/clang/test/SemaCXX/ |
| H A D | overloaded-operator.cpp | 140 struct ArrayLike { struct
|