Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2298
Star 20931
Fork 5422
P
Paddle
未验证 提交 947c5fa7 编写于 作者: C Chang Xu 提交者： GitHub
浏览文件
操作

Fit Quant Dtype (#56019)

上级 7472057c
隐藏空白更改
内联 并排
Showing with 8 addition and 8 deletion
python/paddle/nn/quant/format.py
浏览文件 @ 947c5fa7
...@@ -60,8 +60,8 @@ class LinearQuanter(Layer): ...@@ -60,8 +60,8 @@ class LinearQuanter(Layer):
if in_dynamic_mode(): if in_dynamic_mode():
return _C_ops.quantize_linear( return _C_ops.quantize_linear(
input, input,
self._scales, self._scales.cast(input.dtype),
self._zero_point, self._zero_point.cast(input.dtype),
"quant_axis", "quant_axis",
self._quant_axis, self._quant_axis,
"bit_length", "bit_length",
...@@ -73,8 +73,8 @@ class LinearQuanter(Layer): ...@@ -73,8 +73,8 @@ class LinearQuanter(Layer):
type='quantize_linear', type='quantize_linear',
inputs={ inputs={
'X': input, 'X': input,
'Scale': self._scales, 'Scale': self._scales.cast(input.dtype),
'ZeroPoint': self._zero_point, 'ZeroPoint': self._zero_point.cast(input.dtype),
}, },
outputs={'Y': out}, outputs={'Y': out},
attrs={ attrs={
...@@ -110,8 +110,8 @@ class LinearDequanter(Layer): ...@@ -110,8 +110,8 @@ class LinearDequanter(Layer):
if in_dynamic_mode(): if in_dynamic_mode():
return _C_ops.dequantize_linear( return _C_ops.dequantize_linear(
input, input,
self._scales, self._scales.cast(input.dtype),
self._zero_point, self._zero_point.cast(input.dtype),
"quant_axis", "quant_axis",
self._quant_axis, self._quant_axis,
"bit_length", "bit_length",
...@@ -123,8 +123,8 @@ class LinearDequanter(Layer): ...@@ -123,8 +123,8 @@ class LinearDequanter(Layer):
type='dequantize_linear', type='dequantize_linear',
inputs={ inputs={
'X': input, 'X': input,
'Scale': self._scales, 'Scale': self._scales.cast(input.dtype),
'ZeroPoint': self._zero_point, 'ZeroPoint': self._zero_point.cast(input.dtype),
}, },
outputs={'Y': out}, outputs={'Y': out},
attrs={ attrs={
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或