Skip to content

Commit

Permalink
missing ;
Browse files Browse the repository at this point in the history
  • Loading branch information
prathikr committed Mar 18, 2024
1 parent 771cb5b commit 58912eb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion orttraining/orttraining/core/graph/gradient_builder.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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<int64_t> axes_values = RetrieveValues<int64_t>(attributes.at("axes"));
if (SrcNodeOpsetVersion() < 13) { // axes is attribute for unsqueeze
Expand Down

0 comments on commit 58912eb

Please sign in to comment.