diff --git a/orttraining/orttraining/core/graph/gradient_builder.cc b/orttraining/orttraining/core/graph/gradient_builder.cc index 7576e8814b826..7e8490dfe41af 100755 --- a/orttraining/orttraining/core/graph/gradient_builder.cc +++ b/orttraining/orttraining/core/graph/gradient_builder.cc @@ -1113,7 +1113,7 @@ IMPLEMENT_GRADIENT_BUILDER(GetReduceMeanGradient) { ArgDef grad = GO(0); if (!keepdims) { size_t numInputs = GetSrcNodeInputSize(); - grad = IA("Unqueezed_Grad") + grad = IA("Unqueezed_Grad"); if (attributes.find("axes") != attributes.end()) { std::vector axes_values = RetrieveValues(attributes.at("axes")); if (SrcNodeOpsetVersion() < 13) { // axes is attribute for unsqueeze