MCPcopy
hub / github.com/tinygrad/tinygrad / to_dtype

Method to_dtype

tinygrad/nn/onnx.py:37–37  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

35 UINT64 = 13; BFLOAT16 = 16 # noqa: E702
36
37 def to_dtype(self) -> DType: return DTYPES_DICT[self.name.lower()]
38
39# ***** onnx spec definitions *****
40class Domain(enum.Enum):

Callers 7

_parse_TensorProtoMethod · 0.80
_parse_ValueInfoProtoMethod · 0.80
EyeLikeFunction · 0.80
CastFunction · 0.80
_windowFunction · 0.80
QuantizeLinearFunction · 0.80
_get_expected_dtypeMethod · 0.80

Calls

no outgoing calls

Tested by 1

_get_expected_dtypeMethod · 0.64