Add missing header inclusion

PiperOrigin-RevId: 158265934

1 files changed, 6 insertions, 1 deletions

diff --git a/tensorflow/core/grappler/costs/virtual_scheduler.cc b/tensorflow/core/grappler/costs/virtual_scheduler.cc
index dfa8c768e7..bf352baa45 100644
--- a/tensorflow/core/grappler/costs/virtual_scheduler.cc
+++ b/tensorflow/core/grappler/costs/virtual_scheduler.cc
@@ -14,6 +14,9 @@ limitations under the License.
 ==============================================================================*/
 
 #include "tensorflow/core/grappler/costs/virtual_scheduler.h"
+
+#include <math.h>
+
 #include "tensorflow/core/framework/attr_value.pb.h"
 #include "tensorflow/core/framework/node_def.pb.h"
 #include "tensorflow/core/grappler/clusters/utils.h"
@@ -226,7 +229,9 @@ void VirtualScheduler::MaybeUpdateInputOutput(const NodeDef* node) {
 }
 
 float VirtualScheduler::Round2(const float x) const {
-  return std::round(100.0 * x) / 100.0;
+  // Not using std::round from <cmath> here because not all platforms seem to
+  // support that (specifically Android).
+  return ::round(100.0 * x) / 100.0;
 }
 
 bool VirtualScheduler::IsPersistentNode(const NodeDef* node) const {