Skip to content

Commit

Permalink
Update on "[ONNX] Add binary_cross_entropy_with_logits op to ONNX ops…
Browse files Browse the repository at this point in the history
…et version 12 (#49675)"


Fixes #{#47997}
Exporting the operator binary_cross_entropy_with_logits to ONNX opset version 12.

Differential Revision: [D26050885](https://our.internmc.facebook.com/intern/diff/D26050885)

[ghstack-poisoned]
  • Loading branch information
BowenBao committed Jan 25, 2021
2 parents c0604c8 + fdd6a89 commit 46882b8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/csrc/jit/passes/onnx/shape_type_inference.h
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
#pragma once

#include <torch/csrc/jit/ir/ir.h>
#include <torch/csrc/jit/python/python_arg_flatten.h>
#include <torch/csrc/jit/passes/onnx/helper.h>
#include <torch/csrc/jit/python/python_arg_flatten.h>

namespace torch {
namespace jit {
Expand Down

0 comments on commit 46882b8

Please sign in to comment.