From 6f65e2ad2ce07f57541c97b78b00d2a916c46fd0 Mon Sep 17 00:00:00 2001 From: Derek Murray Date: Thu, 15 Oct 2020 16:54:17 -0700 Subject: [PATCH] Mark the dX and dB outputs of ConvGrad as OpSchema::Optional. (#5462) * Mark the dB output of ConvGrad as OpSchema::Optional. * Also mark dX as optional Co-authored-by: Derek Murray --- orttraining/orttraining/core/graph/training_op_defs.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/orttraining/orttraining/core/graph/training_op_defs.cc b/orttraining/orttraining/core/graph/training_op_defs.cc index 22d3dc7b8f..40729da3b4 100644 --- a/orttraining/orttraining/core/graph/training_op_defs.cc +++ b/orttraining/orttraining/core/graph/training_op_defs.cc @@ -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",