We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2b23874 commit 1b2cde4Copy full SHA for 1b2cde4
core/conversion/evaluators/aten.cpp
@@ -165,7 +165,7 @@ auto aten_registrations = RegisterNodeEvaluators()
165
}).evaluator({
166
c10::Symbol::fromQualString("aten::neg"),
167
[](const torch::jit::Node* n, kwargs& args) -> c10::optional<torch::jit::IValue> {
168
- auto el = args.at(n->input(1)).unwrapToInt();
+ auto el = args.at(n->input(0)).unwrapToInt();
169
170
return el * -1;
171
},
0 commit comments