Skip to content

Commit 3264ea0

Browse files
zonglinpengfacebook-github-bot
authored andcommitted
register quantized_linear.per_tensor in lib (#6563)
Summary: cont of previous diff Differential Revision: D65104400
1 parent b07be36 commit 3264ea0

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

backends/cadence/aot/ops_registrations.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,11 @@
4444
"quantized_linear.out(Tensor src, Tensor weight, Tensor bias, int src_zero_point, Tensor weight_zero_point, Tensor out_multiplier, Tensor out_shift, int out_zero_point, Tensor? offset, *, Tensor(a!) out) -> Tensor(a!)"
4545
)
4646
lib.define(
47-
"cadence::quantized_linear.per_tensor_out(Tensor src, Tensor weight, Tensor bias, SymInt src_zero_point, SymInt weight_zero_point, SymInt out_multiplier, SymInt out_shift, SymInt out_zero_point, Tensor? offset, *, Tensor(a!) out) -> Tensor(a!)"
47+
"quantized_linear.per_tensor_out(Tensor src, Tensor weight, Tensor bias, SymInt src_zero_point, SymInt weight_zero_point, SymInt out_multiplier, SymInt out_shift, SymInt out_zero_point, Tensor? offset, *, Tensor(a!) out) -> Tensor(a!)"
48+
)
49+
lib.define(
50+
"quantized_linear.per_tensor(Tensor src, Tensor weight, Tensor bias, SymInt src_zero_point, "
51+
"SymInt weight_zero_point, SymInt out_multiplier, SymInt out_shift, SymInt out_zero_point, Tensor? offset) -> Tensor"
4852
)
4953

5054
lib.define(

0 commit comments

Comments
 (0)