experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add missing header inclusion

Browse Source 
PiperOrigin-RevId: 158265934
This commit is contained in:
A. Unique TensorFlower 2017-06-07 07:11:36 -07:00 committed by TensorFlower Gardener
parent 3cf88d390c
commit 85355f0159
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/grappler/costs/virtual_scheduler.cc
View File

@ -14,6 +14,9 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "tensorflow/core/grappler/costs/virtual_scheduler.h" #include "tensorflow/core/grappler/costs/virtual_scheduler.h"
#include <math.h>
#include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/framework/attr_value.pb.h"
#include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/framework/node_def.pb.h"
#include "tensorflow/core/grappler/clusters/utils.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 { 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 { bool VirtualScheduler::IsPersistentNode(const NodeDef* node) const {