Skip to content

Commit

Permalink
Update on "[ONNX] Add logical_and, logical_or, logical_xor torch op s…
Browse files Browse the repository at this point in the history
…upport in pytorch exporter (#50570)"

Fixes #{}
Add logical_and, logical_or, logical_xor torch op support in pytorch exporter.

[ghstack-poisoned]
  • Loading branch information
BowenBao committed Jan 22, 2021
2 parents 329bad8 + 5332359 commit 41f3b4f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torch/csrc/jit/passes/onnx/shape_type_inference.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -265,8 +265,8 @@ Node* CloneNodeToGraph(
int opset_version) {
auto vals_to_params_map =
buildValueToParamsMap(n->owningGraph()->block(), params_dict);
auto clone_node =
n_graph->createClone(n, [&n_graph, &vals_to_params_map, opset_version](Value* v) {
auto clone_node = n_graph->createClone(
n, [&n_graph, &vals_to_params_map, opset_version](Value* v) {
auto v_n = v->node();
switch (v_n->kind()) {
case ::c10::onnx::Constant: {
Expand Down

0 comments on commit 41f3b4f

Please sign in to comment.