chore(aten::softmax): Revert redundant change in softmax

narendasan · narendasan · commit 11fada58a8d6 · 2021-02-04T13:49:12.000-08:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/core/conversion/converters/impl/softmax.cpp b/core/conversion/converters/impl/softmax.cpp
@@ -26,9 +26,6 @@ static auto softmax_registrations TRTORCH_UNUSED = RegisterNodeConversionPattern
        }
 
        int64_t dim = args[1].IValue()->toInt();
-       if (dim == -1) {
-         dim = shape.size() - 1;
-       }
        auto softmax = ctx->net->addSoftMax(*in);
 
        TRTORCH_CHECK(softmax, "Unable to create softmax layer from node: " << *n);

Original file line number	Diff line number	Diff line change
`@@ -26,9 +26,6 @@ static auto softmax_registrations TRTORCH_UNUSED = RegisterNodeConversionPattern`
`26`	`26`	`}`
`27`	`27`
`28`	`28`	`int64_t dim = args[1].IValue()->toInt();`
`29`		`- if (dim == -1) {`
`30`		`- dim = shape.size() - 1;`
`31`		`- }`
`32`	`29`	`auto softmax = ctx->net->addSoftMax(*in);`
`33`	`30`
`34`	`31`	`TRTORCH_CHECK(softmax, "Unable to create softmax layer from node: " << *n);`