Skip to content

Commit 6f65e2a

Browse files
Derek MurrayDerek Murray
Derek Murray
and
Derek Murray
authored
Mark the dX and dB outputs of ConvGrad as OpSchema::Optional. (microsoft#5462)
* Mark the dB output of ConvGrad as OpSchema::Optional. * Also mark dX as optional Co-authored-by: Derek Murray <[email protected]>
1 parent 64f6d85 commit 6f65e2a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

orttraining/orttraining/core/graph/training_op_defs.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -418,9 +418,9 @@ void RegisterTrainingOpSchemas() {
418418
.Input(0, "dY", "Gradient of output Y", "T")
419419
.Input(1, "X", "Input tensor", "T")
420420
.Input(2, "W", "Weight tensor", "T")
421-
.Output(0, "dX", "Gradient of input X", "T")
421+
.Output(0, "dX", "Gradient of input X", "T", OpSchema::Optional)
422422
.Output(1, "dW", "Gradient of W", "T")
423-
.Output(2, "dB", "Gradient of B", "T")
423+
.Output(2, "dB", "Gradient of B", "T", OpSchema::Optional)
424424
.AllowUncheckedAttributes()
425425
.TypeConstraint(
426426
"T",

0 commit comments

Comments
 (0)