Searched refs:_make_conversion_method (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | _conversions.py | 40 def _make_conversion_method(name: str, dtype: torch.dtype): function
50 bfloat16 = _make_conversion_method("bfloat16", torch.bfloat16)
52 bool = _make_conversion_method("bool", torch.bool)
54 byte = _make_conversion_method("byte", torch.uint8)
56 cdouble = _make_conversion_method("cdouble", torch.cdouble)
58 cfloat = _make_conversion_method("cfloat", torch.cfloat)
60 chalf = _make_conversion_method("chalf", torch.complex32)
62 char = _make_conversion_method("char", torch.int8)
64 double = _make_conversion_method("double", torch.double)
66 float = _make_conversion_method("float", torch.float)
[all …]
|