diff --git a/test-rt/test-tflite/suite.rs b/test-rt/test-tflite/suite.rs index 3769391037..b943fee21a 100644 --- a/test-rt/test-tflite/suite.rs +++ b/test-rt/test-tflite/suite.rs @@ -170,5 +170,5 @@ fn compatible_conv_q(qcp: &QConvProblem) -> bool { { return true; } - return false; + false } diff --git a/tflite/src/ops/cnn.rs b/tflite/src/ops/cnn.rs index d13f4eb74c..dad4c86ff0 100644 --- a/tflite/src/ops/cnn.rs +++ b/tflite/src/ops/cnn.rs @@ -79,7 +79,7 @@ fn ser_conv( &format!("{node_name}.weights"), &conv.kernel, &vec![k0_tract; conv.kernel.shape()[0]], - &kscale, + kscale, 0, )?); let bscale = kscale.iter().map(|k| k * iscale).collect_vec(); diff --git a/tflite/src/ops/mod.rs b/tflite/src/ops/mod.rs index c11da8fd81..e76047f26e 100644 --- a/tflite/src/ops/mod.rs +++ b/tflite/src/ops/mod.rs @@ -84,9 +84,9 @@ fn linearops_quantization_suport( }; inputs.push(op.ctx.target.add_const(format!("{p}.k0"), k_zp.into_arc_tensor())?); inputs.push(op.ctx.target.add_const(format!("{p}.kscale"), k_scale)?); - inputs.push(op.ctx.target.add_const(format!("{p}.i0"), rctensor0(iqp.zp_scale().0 as i32))?); + inputs.push(op.ctx.target.add_const(format!("{p}.i0"), rctensor0(iqp.zp_scale().0))?); inputs.push(op.ctx.target.add_const(format!("{p}.iscale"), rctensor0(iqp.zp_scale().1))?); - inputs.push(op.ctx.target.add_const(format!("{p}.c0"), rctensor0(oqp.zp_scale().0 as i32))?); + inputs.push(op.ctx.target.add_const(format!("{p}.c0"), rctensor0(oqp.zp_scale().0))?); inputs.push(op.ctx.target.add_const(format!("{p}.cscale"), rctensor0(oqp.zp_scale().1))?); Ok(Some(oqp)) } else {