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
|