Skip to content

P
Paddle

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

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

Make QDQ FP32 (#56059)

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