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.
2 parents bc2a4f5 + 1b2cde4 commit 1168092Copy full SHA for 1168092
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