@@ -213,8 +213,8 @@ auto element_wise_registrations TRTORCH_UNUSED =
213
213
auto unuse = Weights ();
214
214
// IScaleLayer assert shift, scale and power to have
215
215
// the same dtype
216
- auto scaleLayer = ctx-> net -> addScale (
217
- *self, nvinfer1::ScaleMode::kUNIFORM , unuse.data , scaleW.data , unuse.data );
216
+ auto scaleLayer =
217
+ ctx-> net -> addScale ( *self, nvinfer1::ScaleMode::kUNIFORM , unuse.data , scaleW.data , unuse.data );
218
218
TRTORCH_CHECK (scaleLayer, " Unable to create scale layer from node: " << *n);
219
219
self = scaleLayer->getOutput (0 );
220
220
}
@@ -240,8 +240,8 @@ auto element_wise_registrations TRTORCH_UNUSED =
240
240
auto unuse = Weights ();
241
241
// IScaleLayer assert shift, scale and power to have
242
242
// the same dtype
243
- auto scaleLayer = ctx-> net -> addScale (
244
- *self, nvinfer1::ScaleMode::kUNIFORM , unuse.data , scaleW.data , unuse.data );
243
+ auto scaleLayer =
244
+ ctx-> net -> addScale ( *self, nvinfer1::ScaleMode::kUNIFORM , unuse.data , scaleW.data , unuse.data );
245
245
TRTORCH_CHECK (scaleLayer, " Unable to create scale layer from node: " << *n);
246
246
self = scaleLayer->getOutput (0 );
247
247
}
@@ -415,8 +415,8 @@ auto element_wise_registrations TRTORCH_UNUSED =
415
415
// TODO: Remove with functionalization
416
416
auto self = args[0 ].ITensorOrFreeze (ctx);
417
417
auto other = args[1 ].ITensorOrFreeze (ctx);
418
- auto floor_divide =
419
- add_elementwise ( ctx, nvinfer1::ElementWiseOperation::kFLOOR_DIV , self, other, util::node_info (n));
418
+ auto floor_divide = add_elementwise (
419
+ ctx, nvinfer1::ElementWiseOperation::kFLOOR_DIV , self, other, util::node_info (n));
420
420
TRTORCH_CHECK (floor_divide, " Unable to create floor_divide layer from node: " << *n);
421
421
422
422
floor_divide->setName (util::node_info (n).c_str ());
@@ -430,8 +430,8 @@ auto element_wise_registrations TRTORCH_UNUSED =
430
430
auto self = args[0 ].ITensorOrFreeze (ctx);
431
431
auto otherScalar = args[1 ].unwrapToScalar ().to <float >();
432
432
auto other = tensor_to_const (ctx, torch::tensor ({otherScalar}));
433
- auto floor_divide =
434
- add_elementwise ( ctx, nvinfer1::ElementWiseOperation::kFLOOR_DIV , self, other, util::node_info (n));
433
+ auto floor_divide = add_elementwise (
434
+ ctx, nvinfer1::ElementWiseOperation::kFLOOR_DIV , self, other, util::node_info (n));
435
435
TRTORCH_CHECK (floor_divide, " Unable to create floor_divide layer from node: " << *n);
436
436
437
437
floor_divide->setName (util::node_info (n).c_str ());
0 commit comments