Skip to content

Commit 2771a40

Browse files
committed
fixed errors in PR comments
1 parent e598f39 commit 2771a40

File tree

3 files changed

+0
-242
lines changed

3 files changed

+0
-242
lines changed

initializer.py

-231
This file was deleted.

onnxscript/rewriter/onnxruntime/transformers/multihead_attention.py

-6
Original file line numberDiff line numberDiff line change
@@ -52,22 +52,16 @@
5252
import abc
5353
import dataclasses
5454
import logging
55-
import numpy as np
5655
import onnx
5756
from onnx import helper as onnx_helper
58-
from onnx import NodeProto, TensorProto, helper, numpy_helper
5957

6058
import onnxscript
6159
from onnxscript import ir
62-
from onnxscript.ir import SymbolicDim
6360
from onnxscript.rewriter import _ir_utils, function_rule
6461

6562
logger = logging.getLogger(__name__)
6663

67-
import dataclasses
68-
import abc
6964
from onnxscript import ir
70-
from typing import List, Tuple, Union, Optional
7165

7266
from onnxscript.rewriter import _ir_utils, function_rule
7367
import onnx.shape_inference

testingg.py

-5
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,8 @@
44
import onnxscript.optimizer
55
import onnxscript.rewriter
66
from onnxscript.rewriter import onnxruntime as ort_rewriter
7-
from onnx import NodeProto, TensorProto, helper, external_data_helper, numpy_helper
8-
import numpy as np
9-
from onnxscript import ir
107
from onnxscript.ir import convenience as ir_convenience
118

12-
from typing import List, Optional, Tuple, Union
13-
149
def export_to_onnx(input_model_path: str, output_model_path: str):
1510
"""
1611
Export a model to ONNX. This is for testing your rewrite rules without accessing the initializers

0 commit comments

Comments
 (0)