Skip to content
Snippets Groups Projects
Commit 54c3b2b5 authored by auphelia's avatar auphelia
Browse files

[Test] Remove saving points in absorb mul into top k test

parent 74bee9c2
No related branches found
No related tags found
No related merge requests found
......@@ -77,9 +77,7 @@ def test_absorb_mul_into_topk(mul_positive, scalar):
model = model.transform(InferDataTypes())
model = model.transform(GiveUniqueNodeNames())
model = model.transform(GiveReadableTensorNames())
model.save("test.onnx")
model_transformed = model.transform(AbsorbScalarMulIntoTopK())
model_transformed.save("test2.onnx")
# compare execution results
inp_values = np.random.uniform(low=-10, high=10, size=(1, 1, 1, 1000)).astype(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment