Home
last modified time | relevance | path

Searched defs:TensorLikeType (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py475 def _ref(a: TensorLikeType) -> TensorLikeType:
579 def is_complex(input: TensorLikeType):
585 def conj_physical(input: TensorLikeType):
643 def fill(a: TensorLikeType, value: NumberType) -> TensorLikeType:
655 def fill_(a: TensorLikeType, value: NumberType) -> TensorLikeType:
663 def zero(input: TensorLikeType) -> TensorLikeType:
673 def frac(x: TensorLikeType) -> TensorLikeType:
679 def imag(a: TensorLikeType) -> TensorLikeType:
691 def isfinite(a: TensorLikeType) -> TensorLikeType:
699 def isinf(a: TensorLikeType) -> TensorLikeType:
[all …]
H A Dfft.py43 x: TensorLikeType, norm: NormType, signal_numel: int, forward: bool
82 t: TensorLikeType, require_complex: bool = False
91 x: TensorLikeType, dims: Tuple[int, ...], sizes: Tuple[int, ...]
118 input: TensorLikeType,
145 input: TensorLikeType,
174 input: TensorLikeType,
201 input: TensorLikeType,
215 input: TensorLikeType,
229 input: TensorLikeType,
240 input: TensorLikeType,
[all …]
H A D_conversions.py42 self: TensorLikeType, memory_format: torch.memory_format = torch.preserve_format
81 def complex(real: TensorLikeType, imag: TensorLikeType) -> TensorLikeType:
115 def polar(abs: TensorLikeType, angle: TensorLikeType) -> TensorLikeType:
/aosp_15_r20/external/pytorch/torch/_refs/nn/functional/
H A D__init__.py76 self: TensorLikeType,
97 self: TensorLikeType, p: float = 0.5, training: bool = False, inplace: bool = False
162 a: TensorLikeType, alpha: Optional[NumberType] = None, inplace: bool = False
187 a: TensorLikeType, p: float = 0.5, training: bool = True, inplace: bool = False
220 a: TensorLikeType,
257 def relu(a: TensorLikeType, inplace: bool = False) -> TensorLikeType:
270 def channel_shuffle(input: TensorLikeType, groups: int) -> TensorLikeType:
367 a: TensorLikeType, negative_slope: float = 0.01, inplace: bool = False
390 def mish(a: TensorLikeType, inplace: bool = False) -> TensorLikeType:
407 def selu(a: TensorLikeType, inplace: bool = False) -> TensorLikeType:
[all …]
/aosp_15_r20/external/pytorch/torch/_refs/special/
H A D__init__.py52 def bessel_j0(a: TensorLikeType) -> TensorLikeType:
59 def bessel_j1(a: TensorLikeType) -> TensorLikeType:
69 def entr(a: TensorLikeType) -> TensorLikeType:
83 def erfcx(a: TensorLikeType) -> TensorLikeType:
94 def i0e(a: TensorLikeType) -> TensorLikeType:
101 def i1(a: TensorLikeType) -> TensorLikeType:
108 def i1e(a: TensorLikeType) -> TensorLikeType:
118 def log_ndtr(a: TensorLikeType) -> TensorLikeType:
135 def logit(self: TensorLikeType, eps: Optional[float] = None) -> TensorLikeType:
150 def xlog1py(a: Union[TensorLikeType, NumberType], b: Union[TensorLikeType, NumberType]):
[all …]
/aosp_15_r20/external/pytorch/torch/_prims/
H A D__init__.py385 args_with_fixed_dtypes: Optional[Tuple[TensorLikeType, ...]] = None,
665 def _conj_physical_meta(input: TensorLikeType) -> TensorLikeType:
683 input: TensorLikeType, *, memory_format: torch.memory_format = torch.preserve_format
771 def _fill_meta(a: TensorLikeType, value: NumberType) -> TensorLikeType:
1110 a: Union[TensorLikeType, NumberType], b: Union[TensorLikeType, NumberType]
1129 a: Union[TensorLikeType, NumberType], b: Union[TensorLikeType, NumberType]
1216 a: TensorLikeType, size: ShapeType, stride: StrideType, storage_offset: int
1256 a: TensorLikeType, shape: ShapeType, broadcast_dimensions: Sequence[int]
1375 a: TensorLikeType, start: int, end: int
1432 def _collapse_view_meta(a: TensorLikeType, start: int, end: int) -> TensorLikeType:
[all …]
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py91 TensorLikeType = torch.Tensor variable
137 a: TensorLikeType,
138 b: TensorLikeType,
197 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True, significant_only=True
215 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True
221 a: TensorLikeType, b: TensorLikeType, *, only_cuda=True
227 def is_contiguous(a: TensorLikeType) -> bool:
1089 def check_is_matrix(A: TensorLikeType, f_name: str, arg_name: str = "A"):
1123 a: Optional[Union[torch.dtype, TensorLikeType, NumberType]], argument
1136 x: Optional[Union[torch.dtype, TensorLikeType, NumberType]] argument
[all …]
H A Dwrappers.py28 def _maybe_convert_to_dtype(a: TensorLikeType, dtype: torch.dtype) -> TensorLikeType:
160 def _resize_output_check(out: TensorLikeType, shape: ShapeType):
178 out: TensorLikeType,
188 def is_cpu_scalar(x: TensorLikeType) -> bool:
193 *, copy_from: TensorLikeType, copy_to: TensorLikeType, exact_dtype: bool = False
/aosp_15_r20/external/pytorch/torch/_refs/linalg/
H A D__init__.py91 input: TensorLikeType,
103 x: TensorLikeType,
175 A: TensorLikeType,
257 A: TensorLikeType,
291 def svd(A: TensorLikeType, full_matrices: bool = True) -> Tuple[Tensor, Tensor, Tensor]:
297 def svdvals(A: TensorLikeType) -> Tensor: