Skip to content

Commit b60fe0c

Browse files
committed
chore: Fix linter issues
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 84f626a commit b60fe0c

File tree

1 file changed

+10
-10
lines changed
  • core/conversion/converters/impl

1 file changed

+10
-10
lines changed

core/conversion/converters/impl/max.cpp

+10-10
Original file line numberDiff line numberDiff line change
@@ -31,18 +31,18 @@ auto max_registrations TORCHTRT_UNUSED = RegisterNodeConversionPatterns().patter
3131

3232
nvinfer1::ITensor* out0;
3333
nvinfer1::ITensor* out1;
34-
if (!keep_dims){
34+
if (!keep_dims) {
3535
if (topk_dims[dim] == 1) {
36-
auto squeeze_layer = ctx->net->addShuffle(*topk_layer->getOutput(0));
37-
squeeze_layer->setReshapeDimensions(util::squeezeDims(topk_layer->getOutput(0)->getDimensions(), dim));
38-
TORCHTRT_CHECK(squeeze_layer, "Unable to create squeeze_layer layer from node: " << *n);
39-
out0 = ctx->AssociateValueAndTensor(n->outputs()[0], squeeze_layer->getOutput(0));
40-
41-
auto squeeze_layer_indices = ctx->net->addShuffle(*topk_layer->getOutput(1));
42-
squeeze_layer_indices->setReshapeDimensions(util::squeezeDims(topk_layer->getOutput(1)->getDimensions(), dim));
43-
TORCHTRT_CHECK(squeeze_layer_indices, "Unable to create squeeze_layer_indices layer from node: " << *n);
44-
out1 = ctx->AssociateValueAndTensor(n->outputs()[1], squeeze_layer_indices->getOutput(0));
36+
auto squeeze_layer = ctx->net->addShuffle(*topk_layer->getOutput(0));
37+
squeeze_layer->setReshapeDimensions(util::squeezeDims(topk_layer->getOutput(0)->getDimensions(), dim));
38+
TORCHTRT_CHECK(squeeze_layer, "Unable to create squeeze_layer layer from node: " << *n);
39+
out0 = ctx->AssociateValueAndTensor(n->outputs()[0], squeeze_layer->getOutput(0));
4540

41+
auto squeeze_layer_indices = ctx->net->addShuffle(*topk_layer->getOutput(1));
42+
squeeze_layer_indices->setReshapeDimensions(
43+
util::squeezeDims(topk_layer->getOutput(1)->getDimensions(), dim));
44+
TORCHTRT_CHECK(squeeze_layer_indices, "Unable to create squeeze_layer_indices layer from node: " << *n);
45+
out1 = ctx->AssociateValueAndTensor(n->outputs()[1], squeeze_layer_indices->getOutput(0));
4646
}
4747
} else {
4848
out0 = ctx->AssociateValueAndTensor(n->outputs()[0], topk_layer->getOutput(0));

0 commit comments

Comments
 (0)