Searched defs:TensorOrNumberLikeType (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 1863 min: Optional[TensorOrNumberLikeType] = None,
1864 max: Optional[TensorOrNumberLikeType] = None,
1891 min: Optional[TensorOrNumberLikeType] = None,
1900 max: Optional[TensorOrNumberLikeType] = None,
1920 a: Optional[TensorOrNumberLikeType] = None,
1921 b: Optional[TensorOrNumberLikeType] = None,
5593 def masked_fill(a: TensorLikeType, mask: TensorLikeType, value: TensorOrNumberLikeType):
5639 a: TensorLikeType, mask: TensorLikeType, value: TensorOrNumberLikeType
5914 a: TensorOrNumberLikeType,
6426 scalarType: torch.dtype, obj: Union[TensorOrNumberLikeType, TensorSequenceType]
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 94 TensorOrNumberLikeType: TypeAlias = Union[TensorLikeType, NumberType] variable
|