1818"""Dense alter op functions for ARM"""
1919
2020import tvm
21- from tvm import te
2221from tvm import relay
23- from tvm import autotvm
24- from ..utils import get_const_tuple
2522from .. import nn
2623from ..nn import dense_alter_layout
2724
@@ -137,7 +134,7 @@ def _dense_legalize(attrs, inputs, arg_types):
137134 tvm .ir .tensor_type .TensorType ([N + dn , arg_types [1 ].shape [1 ]], arg_types [1 ].dtype ),
138135 ]
139136
140- vrmpy_out = vrmpy_legalize (x , y_ , arg_types , relay .nn .dense , new_attrs , False )
137+ vrmpy_out = vrmpy_legalize (x , y_ , arg_types , relay .nn .dense , new_attrs )
141138
142139 if vrmpy_out is None :
143140 out_ = relay .nn .dense (x , y_ , ** new_attrs )
@@ -147,4 +144,4 @@ def _dense_legalize(attrs, inputs, arg_types):
147144 out = relay .strided_slice (out_ , begin = [0 , 0 ], end = [x .value for x in output_tensor .shape ])
148145 return out
149146
150- return vrmpy_legalize (inputs [0 ], inputs [1 ], arg_types , relay .nn .dense , attrs , False )
147+ return vrmpy_legalize (inputs [0 ], inputs [1 ], arg_types , relay .nn .dense , attrs )
0 commit comments