| /external/pytorch/torch/_refs/special/ |
| D | __init__.py | 16 TensorLikeType,
52 def bessel_j0(a: TensorLikeType) -> TensorLikeType: argument
59 def bessel_j1(a: TensorLikeType) -> TensorLikeType: argument
69 def entr(a: TensorLikeType) -> TensorLikeType: argument
83 def erfcx(a: TensorLikeType) -> TensorLikeType: argument
94 def i0e(a: TensorLikeType) -> TensorLikeType: argument
101 def i1(a: TensorLikeType) -> TensorLikeType: argument
108 def i1e(a: TensorLikeType) -> TensorLikeType: argument
118 def log_ndtr(a: TensorLikeType) -> TensorLikeType: argument
135 def logit(self: TensorLikeType, eps: Optional[float] = None) -> TensorLikeType: argument
[all …]
|
| /external/pytorch/torch/_refs/nn/functional/ |
| D | __init__.py | 17 TensorLikeType,
76 self: TensorLikeType, argument
78 ) -> TensorLikeType:
97 self: TensorLikeType, p: float = 0.5, training: bool = False, inplace: bool = False argument
98 ) -> TensorLikeType:
162 a: TensorLikeType, alpha: Optional[NumberType] = None, inplace: bool = False argument
163 ) -> TensorLikeType:
171 rhs: TensorLikeType
187 a: TensorLikeType, p: float = 0.5, training: bool = True, inplace: bool = False argument
188 ) -> TensorLikeType:
[all …]
|
| /external/pytorch/torch/_refs/ |
| D | fft.py | 8 from torch._prims_common import DimsType, ShapeType, TensorLikeType
43 x: TensorLikeType, norm: NormType, signal_numel: int, forward: bool argument
44 ) -> TensorLikeType:
82 t: TensorLikeType, require_complex: bool = False argument
83 ) -> TensorLikeType:
91 x: TensorLikeType, dims: Tuple[int, ...], sizes: Tuple[int, ...] argument
92 ) -> TensorLikeType:
118 input: TensorLikeType, argument
123 ) -> TensorLikeType:
145 input: TensorLikeType, argument
[all …]
|
| D | __init__.py | 39 TensorLikeType,
475 def _ref(a: TensorLikeType) -> TensorLikeType: argument
579 def is_complex(input: TensorLikeType): argument
585 def conj_physical(input: TensorLikeType): argument
643 def fill(a: TensorLikeType, value: NumberType) -> TensorLikeType: argument
655 def fill_(a: TensorLikeType, value: NumberType) -> TensorLikeType: argument
663 def zero(input: TensorLikeType) -> TensorLikeType: argument
673 def frac(x: TensorLikeType) -> TensorLikeType: argument
679 def imag(a: TensorLikeType) -> TensorLikeType: argument
691 def isfinite(a: TensorLikeType) -> TensorLikeType: argument
[all …]
|
| D | _conversions.py | 7 from torch._prims_common import TensorLikeType
42 self: TensorLikeType, memory_format: torch.memory_format = torch.preserve_format argument
43 ) -> TensorLikeType:
81 def complex(real: TensorLikeType, imag: TensorLikeType) -> TensorLikeType: argument
115 def polar(abs: TensorLikeType, angle: TensorLikeType) -> TensorLikeType: argument
|
| /external/pytorch/torch/_prims/ |
| D | __init__.py | 27 TensorLikeType,
385 args_with_fixed_dtypes: Optional[Tuple[TensorLikeType, ...]] = None, argument
665 def _conj_physical_meta(input: TensorLikeType) -> TensorLikeType: argument
683 input: TensorLikeType, *, memory_format: torch.memory_format = torch.preserve_format argument
684 ) -> TensorLikeType:
771 def _fill_meta(a: TensorLikeType, value: NumberType) -> TensorLikeType: argument
1110 a: Union[TensorLikeType, NumberType], b: Union[TensorLikeType, NumberType] argument
1111 ) -> TensorLikeType:
1129 a: Union[TensorLikeType, NumberType], b: Union[TensorLikeType, NumberType] argument
1130 ) -> TensorLikeType:
[all …]
|
| /external/pytorch/torch/_prims_common/ |
| D | wrappers.py | 17 TensorLikeType,
28 def _maybe_convert_to_dtype(a: TensorLikeType, dtype: torch.dtype) -> TensorLikeType: argument
160 def _resize_output_check(out: TensorLikeType, shape: ShapeType): argument
178 out: TensorLikeType, argument
188 def is_cpu_scalar(x: TensorLikeType) -> bool: argument
193 *, copy_from: TensorLikeType, copy_to: TensorLikeType, exact_dtype: bool = False argument
247 TensorLikeType
249 else Tuple[tuple(TensorLikeType for _ in range(len(out_names)))]
252 TensorLikeType
255 f"return_types_{fn.__name__}", [(o, TensorLikeType) for o in out_names]
|
| D | __init__.py | 91 TensorLikeType = torch.Tensor variable
93 TensorSequenceType: TypeAlias = Union[List[TensorLikeType], Tuple[TensorLikeType, ...]]
94 TensorOrNumberLikeType: TypeAlias = Union[TensorLikeType, NumberType]
137 a: TensorLikeType, argument
138 b: TensorLikeType, argument
197 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True, significant_only=True argument
215 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True argument
221 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True argument
227 def is_contiguous(a: TensorLikeType) -> bool: argument
1089 def check_is_matrix(A: TensorLikeType, f_name: str, arg_name: str = "A"): argument
[all …]
|
| /external/pytorch/torch/_refs/linalg/ |
| D | __init__.py | 18 TensorLikeType,
91 input: TensorLikeType, argument
96 ) -> TensorLikeType:
103 x: TensorLikeType, argument
175 A: TensorLikeType, argument
181 ) -> TensorLikeType:
257 A: TensorLikeType, argument
263 ) -> TensorLikeType:
291 def svd(A: TensorLikeType, full_matrices: bool = True) -> Tuple[Tensor, Tensor, Tensor]: argument
297 def svdvals(A: TensorLikeType) -> Tensor: argument
|