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);