@@ -160,7 +160,7 @@ def aten_ops_gelu(
160
160
kwargs : Dict [str , Argument ],
161
161
name : str ,
162
162
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
163
- return impl .activation .gelu (
163
+ return impl .actv .gelu (
164
164
network ,
165
165
target ,
166
166
SourceIR .ATEN ,
@@ -177,7 +177,7 @@ def aten_ops_relu(
177
177
kwargs : Dict [str , Argument ],
178
178
name : str ,
179
179
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
180
- return impl .activation .relu (
180
+ return impl .actv .relu (
181
181
network ,
182
182
target ,
183
183
SourceIR .ATEN ,
@@ -194,7 +194,7 @@ def aten_ops_sigmoid(
194
194
kwargs : Dict [str , Argument ],
195
195
name : str ,
196
196
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
197
- return impl .activation .sigmoid (
197
+ return impl .actv .sigmoid (
198
198
network ,
199
199
target ,
200
200
SourceIR .ATEN ,
@@ -211,7 +211,7 @@ def aten_ops_tanh(
211
211
kwargs : Dict [str , Argument ],
212
212
name : str ,
213
213
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
214
- return impl .activation .tanh (
214
+ return impl .actv .tanh (
215
215
network ,
216
216
target ,
217
217
SourceIR .ATEN ,
@@ -228,7 +228,7 @@ def aten_ops_leaky_relu(
228
228
kwargs : Dict [str , Argument ],
229
229
name : str ,
230
230
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
231
- return impl .activation .leaky_relu (
231
+ return impl .actv .leaky_relu (
232
232
network ,
233
233
target ,
234
234
SourceIR .ATEN ,
@@ -246,7 +246,7 @@ def aten_ops_elu(
246
246
kwargs : Dict [str , Argument ],
247
247
name : str ,
248
248
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
249
- return impl .activation .elu (
249
+ return impl .actv .elu (
250
250
network ,
251
251
target ,
252
252
SourceIR .ATEN ,
@@ -265,7 +265,7 @@ def aten_ops_softplus(
265
265
kwargs : Dict [str , Argument ],
266
266
name : str ,
267
267
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
268
- return impl .activation .softplus (
268
+ return impl .actv .softplus (
269
269
network ,
270
270
target ,
271
271
SourceIR .ATEN ,
@@ -283,7 +283,7 @@ def aten_ops_clip(
283
283
kwargs : Dict [str , Argument ],
284
284
name : str ,
285
285
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
286
- return impl .activation .clip (
286
+ return impl .actv .clip (
287
287
network ,
288
288
target ,
289
289
SourceIR .ATEN ,
@@ -302,7 +302,7 @@ def aten_ops_hard_sigmoid(
302
302
kwargs : Dict [str , Argument ],
303
303
name : str ,
304
304
) -> Union [TRTTensor , Sequence [TRTTensor ]]:
305
- return impl .activation .hard_sigmoid (
305
+ return impl .actv .hard_sigmoid (
306
306
network ,
307
307
target ,
308
308
SourceIR .ATEN ,
0 commit comments