File tree 2 files changed +2
-2
lines changed
optimum/intel/neural_compressor
2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change 25
25
import torch
26
26
from datasets import Dataset , load_dataset
27
27
from neural_compressor .config import PostTrainingQuantConfig
28
- from neural_compressor .experimental .export import torch_to_int8_onnx
28
+ from neural_compressor .utils .export import torch_to_int8_onnx
29
29
from neural_compressor .model .onnx_model import ONNXModel
30
30
from neural_compressor .model .torch_model import IPEXModel , PyTorchModel
31
31
from neural_compressor .quantization import fit
Original file line number Diff line number Diff line change 36
36
from neural_compressor import training
37
37
from neural_compressor .compression import DistillationCallbacks
38
38
from neural_compressor .conf .pythonic_config import _BaseQuantizationConfig
39
- from neural_compressor .experimental .export import torch_to_fp32_onnx , torch_to_int8_onnx
39
+ from neural_compressor .utils .export import torch_to_fp32_onnx , torch_to_int8_onnx
40
40
from packaging import version
41
41
from torch import nn
42
42
from torch .utils .data import Dataset , RandomSampler
You can’t perform that action at this time.
0 commit comments