Skip to content

P
Paddle

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

通知 2297
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 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):
if in_dynamic_mode():
return _C_ops.quantize_linear(
input,
self._scales,
self._zero_point,
self._scales.cast(input.dtype),
self._zero_point.cast(input.dtype),
"quant_axis",
self._quant_axis,
"bit_length",
......@@ -73,8 +73,8 @@ class LinearQuanter(Layer):
type='quantize_linear',
inputs={
'X': input,
'Scale': self._scales,
'ZeroPoint': self._zero_point,
'Scale': self._scales.cast(input.dtype),
'ZeroPoint': self._zero_point.cast(input.dtype),
},
outputs={'Y': out},
attrs={
......@@ -110,8 +110,8 @@ class LinearDequanter(Layer):
if in_dynamic_mode():
return _C_ops.dequantize_linear(
input,
self._scales,
self._zero_point,
self._scales.cast(input.dtype),
self._zero_point.cast(input.dtype),
"quant_axis",
self._quant_axis,
"bit_length",
......@@ -123,8 +123,8 @@ class LinearDequanter(Layer):
type='dequantize_linear',
inputs={
'X': input,
'Scale': self._scales,
'ZeroPoint': self._zero_point,
'Scale': self._scales.cast(input.dtype),
'ZeroPoint': self._zero_point.cast(input.dtype),
},
outputs={'Y': out},
attrs={
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或