We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bc24f2f commit 875a869Copy full SHA for 875a869
neural_compressor/experimental/export/utils.py
@@ -64,7 +64,7 @@ def __init__(self, fp32_onnx_path):
64
for dim in node.shape:
65
shape.append(dim if isinstance(dim, int) else 1)
66
dtype = ONNX2Numpy_dtype(node.type)
67
- input[node.name] = np.zeros(shape).astype(dtype)
+ input[node.name] = np.ones(shape).astype(dtype)
68
self.data = [input]
69
self.data = iter(self.data)
70
0 commit comments