Skip to content

Commit 5c7bfad

Browse files
authored
Xu 113 120 rc3 fix func name (#2532)
* fix CPU UTs * rename function.
1 parent b8848b5 commit 5c7bfad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

intel_extension_for_pytorch/quantization/_quantize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,9 @@ def prepare(
3131
"""
3232
assert isinstance(model, torch.nn.Module), "Only support nn.Module prepare for quantization path"
3333
# auto model channels_last memory format conversion
34-
from ..frontend import auto_channels_last, _convert_convNd_weight_memory_format
34+
from ..frontend import auto_channels_last, _convert_convNd_deconvNd_weight_memory_format
3535
if auto_channels_last:
36-
_convert_convNd_weight_memory_format(model)
36+
_convert_convNd_deconvNd_weight_memory_format(model)
3737
try:
3838
prepare_model = optimization.fuse(model, inplace=inplace)
3939
prepare_model = linear_bn_fuse(prepare_model, inplace=inplace)

0 commit comments

Comments
 (0)