Skip to content

Commit 5a49077

Browse files
committed
Remove impl:: namespace from scalar_to_tensor calls
1 parent cb1f504 commit 5a49077

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

core/conversion/converters/impl/element_wise.cpp

100644100755
Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ auto element_wise_registrations TORCHTRT_UNUSED =
5858
auto scalar = args[2].unwrapToScalar();
5959

6060
if (1 != scalar.to<float>()) {
61-
auto alphaTensor = impl::scalar_to_tensor(ctx, scalar);
61+
auto alphaTensor = scalar_to_tensor(ctx, scalar);
6262
auto scaleLayer = add_elementwise(
6363
ctx,
6464
nvinfer1::ElementWiseOperation::kPROD,
@@ -87,7 +87,7 @@ auto element_wise_registrations TORCHTRT_UNUSED =
8787
auto scalar = args[2].unwrapToScalar();
8888

8989
if (1 != scalar.to<float>()) {
90-
auto alphaTensor = impl::scalar_to_tensor(ctx, scalar);
90+
auto alphaTensor = scalar_to_tensor(ctx, scalar);
9191
auto scaleLayer = add_elementwise(
9292
ctx,
9393
nvinfer1::ElementWiseOperation::kPROD,
@@ -262,11 +262,11 @@ auto element_wise_registrations TORCHTRT_UNUSED =
262262
[](ConversionCtx* ctx, const torch::jit::Node* n, args& args) -> bool {
263263
// Should implement other - alpha * self
264264
auto self = args[0].ITensorOrFreeze(ctx);
265-
auto other = impl::scalar_to_tensor(ctx, args[1].unwrapToScalar());
265+
auto other = scalar_to_tensor(ctx, args[1].unwrapToScalar());
266266
auto scalar = args[2].unwrapToScalar();
267267

268268
if (1 != scalar.to<float>()) {
269-
auto alphaTensor = impl::scalar_to_tensor(ctx, scalar);
269+
auto alphaTensor = scalar_to_tensor(ctx, scalar);
270270
auto scaleLayer = add_elementwise(
271271
ctx,
272272
nvinfer1::ElementWiseOperation::kPROD,
@@ -294,7 +294,7 @@ auto element_wise_registrations TORCHTRT_UNUSED =
294294
auto scalar = args[2].unwrapToScalar();
295295

296296
if (1 != scalar.to<float>()) {
297-
auto alphaTensor = impl::scalar_to_tensor(ctx, scalar);
297+
auto alphaTensor = scalar_to_tensor(ctx, scalar);
298298
auto scaleLayer = add_elementwise(
299299
ctx,
300300
nvinfer1::ElementWiseOperation::kPROD,

0 commit comments

Comments
 (0)