From b9077f91f4c1462b37b4c03bfb5aa8e1d8908933 Mon Sep 17 00:00:00 2001
From: Gauri1 Deshpande <gauri1.deshpande@intel.com>
Date: Thu, 18 Apr 2024 12:41:41 -0700
Subject: [PATCH] add ops back to MklList for auto_mixed_precision;
 accidentally removed in another PR

---
 .../core/grappler/optimizers/auto_mixed_precision_lists.h       | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h b/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h
index fafce179906237..a4bc290c18be46 100644
--- a/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h
+++ b/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h
@@ -442,6 +442,7 @@ class AutoMixedPrecisionListsMkl : public AutoMixedPrecisionLists {
                                      "Log",
                                      "Log1p",
                                      "LogSoftmax",
+                                     "Mean",
                                      "Prod",
                                      "RealDiv",
                                      "Reciprocal",
@@ -458,6 +459,7 @@ class AutoMixedPrecisionListsMkl : public AutoMixedPrecisionLists {
                                      "Sqrt",
                                      "Square",
                                      "SquaredDifference",
+                                     "Sum",
                                      "Tanh",
                                      "TanhGrad"};
     UpdateList("INFERLIST", &list);