Searched refs:TensorOrNumberLikeType (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_refs/ |
| D | __init__.py | 40 TensorOrNumberLikeType,
1863 min: Optional[TensorOrNumberLikeType] = None, argument
1864 max: Optional[TensorOrNumberLikeType] = None, argument
1891 min: Optional[TensorOrNumberLikeType] = None, argument
1900 max: Optional[TensorOrNumberLikeType] = None, argument
1920 a: Optional[TensorOrNumberLikeType] = None, argument
1921 b: Optional[TensorOrNumberLikeType] = None, argument
5593 def masked_fill(a: TensorLikeType, mask: TensorLikeType, value: TensorOrNumberLikeType): argument
5639 a: TensorLikeType, mask: TensorLikeType, value: TensorOrNumberLikeType argument
5914 a: TensorOrNumberLikeType, argument
[all …]
|
| /external/pytorch/torch/_prims_common/ |
| D | __init__.py | 94 TensorOrNumberLikeType: TypeAlias = Union[TensorLikeType, NumberType] variable
|