@@ -160,7 +160,7 @@ def aten_ops_gelu(
160160 kwargs : Dict [str , Argument ],
161161 name : str ,
162162) -> Union [TRTTensor , Sequence [TRTTensor ]]:
163- return impl .activation .gelu (
163+ return impl .actv .gelu (
164164 network ,
165165 target ,
166166 SourceIR .ATEN ,
@@ -177,7 +177,7 @@ def aten_ops_relu(
177177 kwargs : Dict [str , Argument ],
178178 name : str ,
179179) -> Union [TRTTensor , Sequence [TRTTensor ]]:
180- return impl .activation .relu (
180+ return impl .actv .relu (
181181 network ,
182182 target ,
183183 SourceIR .ATEN ,
@@ -194,7 +194,7 @@ def aten_ops_sigmoid(
194194 kwargs : Dict [str , Argument ],
195195 name : str ,
196196) -> Union [TRTTensor , Sequence [TRTTensor ]]:
197- return impl .activation .sigmoid (
197+ return impl .actv .sigmoid (
198198 network ,
199199 target ,
200200 SourceIR .ATEN ,
@@ -211,7 +211,7 @@ def aten_ops_tanh(
211211 kwargs : Dict [str , Argument ],
212212 name : str ,
213213) -> Union [TRTTensor , Sequence [TRTTensor ]]:
214- return impl .activation .tanh (
214+ return impl .actv .tanh (
215215 network ,
216216 target ,
217217 SourceIR .ATEN ,
@@ -228,7 +228,7 @@ def aten_ops_leaky_relu(
228228 kwargs : Dict [str , Argument ],
229229 name : str ,
230230) -> Union [TRTTensor , Sequence [TRTTensor ]]:
231- return impl .activation .leaky_relu (
231+ return impl .actv .leaky_relu (
232232 network ,
233233 target ,
234234 SourceIR .ATEN ,
@@ -246,7 +246,7 @@ def aten_ops_elu(
246246 kwargs : Dict [str , Argument ],
247247 name : str ,
248248) -> Union [TRTTensor , Sequence [TRTTensor ]]:
249- return impl .activation .elu (
249+ return impl .actv .elu (
250250 network ,
251251 target ,
252252 SourceIR .ATEN ,
@@ -265,7 +265,7 @@ def aten_ops_softplus(
265265 kwargs : Dict [str , Argument ],
266266 name : str ,
267267) -> Union [TRTTensor , Sequence [TRTTensor ]]:
268- return impl .activation .softplus (
268+ return impl .actv .softplus (
269269 network ,
270270 target ,
271271 SourceIR .ATEN ,
@@ -283,7 +283,7 @@ def aten_ops_clip(
283283 kwargs : Dict [str , Argument ],
284284 name : str ,
285285) -> Union [TRTTensor , Sequence [TRTTensor ]]:
286- return impl .activation .clip (
286+ return impl .actv .clip (
287287 network ,
288288 target ,
289289 SourceIR .ATEN ,
@@ -302,7 +302,7 @@ def aten_ops_hard_sigmoid(
302302 kwargs : Dict [str , Argument ],
303303 name : str ,
304304) -> Union [TRTTensor , Sequence [TRTTensor ]]:
305- return impl .activation .hard_sigmoid (
305+ return impl .actv .hard_sigmoid (
306306 network ,
307307 target ,
308308 SourceIR .ATEN ,
0 commit comments