saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-04 04:07:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Mark the dX and dB outputs of ConvGrad as OpSchema::Optional. (#5462)

Browse source 
* Mark the dB output of ConvGrad as OpSchema::Optional.

* Also mark dX as optional

Co-authored-by: Derek Murray <demurra@microsoft.com>
This commit is contained in:
Derek Murray 2020-10-15 16:54:17 -07:00 committed by GitHub
parent 64f6d856e4
commit 6f65e2ad2c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
orttraining/orttraining/core/graph/training_op_defs.cc
View file

@ -418,9 +418,9 @@ void RegisterTrainingOpSchemas() {
.Input(0, "dY", "Gradient of output Y", "T")
.Input(1, "X", "Input tensor", "T")
.Input(2, "W", "Weight tensor", "T")
.Output(0, "dX", "Gradient of input X", "T")
.Output(0, "dX", "Gradient of input X", "T", OpSchema::Optional)
.Output(1, "dW", "Gradient of W", "T")
.Output(2, "dB", "Gradient of B", "T")
.Output(2, "dB", "Gradient of B", "T", OpSchema::Optional)
.AllowUncheckedAttributes()
.TypeConstraint(
"T",