We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f5f6969 commit 0ce221eCopy full SHA for 0ce221e
backends/cadence/aot/ops_registrations.py
@@ -111,7 +111,7 @@ def quantized_linear_meta(
111
weight_size = list(weight.size())
112
assert len(weight_size) == 2
113
out_size[-1] = weight_size[0]
114
- return src.new_empty(out_size, dtype=torch.uint8)
+ return src.new_empty(out_size, dtype=src.dtype)
115
116
117
@register_fake("cadence::quantized_conv")
0 commit comments