From 1fa1c51544ad60e7779dd809e5ae7b00b6bdd67a Mon Sep 17 00:00:00 2001 From: Vincent Wang Date: Fri, 30 Oct 2020 07:08:19 +0800 Subject: [PATCH] bug fix for name of gradient constant (#5626) Co-authored-by: Vincent Wang --- orttraining/orttraining/core/graph/gradient_builder_base.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/orttraining/orttraining/core/graph/gradient_builder_base.h b/orttraining/orttraining/core/graph/gradient_builder_base.h index 10eab0412a..7822d1ec11 100644 --- a/orttraining/orttraining/core/graph/gradient_builder_base.h +++ b/orttraining/orttraining/core/graph/gradient_builder_base.h @@ -221,15 +221,15 @@ class GradientBuilderBase { } static NodeDef ZeroConstantNode(int elem_type) { - return ConstantScalarNode(0.0f, "ZeroConstant", elem_type); + return ConstantScalarNode(0.0f, "ZeroConstant_Type" + std::to_string(elem_type), elem_type); } static NodeDef HalfConstantNode(int elem_type) { - return ConstantScalarNode(0.5f, "HalfConstant", elem_type); + return ConstantScalarNode(0.5f, "HalfConstant_Type" + std::to_string(elem_type), elem_type); } static NodeDef OneConstantNode(int elem_type) { - return ConstantScalarNode(1.0f, "OneConstant", elem_type); + return ConstantScalarNode(1.0f, "OneConstant_Type" + std::to_string(elem_type), elem_type); } void HandleBroadcasting(const ArgDef& input_grad,