aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/optimizers/constant_folding.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/grappler/optimizers/constant_folding.h')
-rw-r--r--tensorflow/core/grappler/optimizers/constant_folding.h6
1 files changed, 1 insertions, 5 deletions
diff --git a/tensorflow/core/grappler/optimizers/constant_folding.h b/tensorflow/core/grappler/optimizers/constant_folding.h
index dd988f336c..30d778789a 100644
--- a/tensorflow/core/grappler/optimizers/constant_folding.h
+++ b/tensorflow/core/grappler/optimizers/constant_folding.h
@@ -22,7 +22,6 @@ limitations under the License.
#include "tensorflow/core/grappler/costs/graph_properties.h"
#include "tensorflow/core/grappler/optimizers/graph_optimizer.h"
#include "tensorflow/core/grappler/utils.h"
-#include "tensorflow/core/protobuf/rewriter_config.pb.h"
namespace tensorflow {
namespace grappler {
@@ -38,7 +37,6 @@ class ConstantFolding : public GraphOptimizer {
NodeMap* node_map);
ConstantFolding(DeviceBase* cpu_device);
- ConstantFolding(RewriterConfig::Toggle opt_level, DeviceBase* cpu_device);
~ConstantFolding() override {}
@@ -53,8 +51,7 @@ class ConstantFolding : public GraphOptimizer {
private:
Status MaterializeShapes(const GrapplerItem& item,
const GraphProperties& properties);
- Status MaterializeConstants(const GrapplerItem& item,
- const GraphProperties& properties);
+
bool IsFoldable(const NodeDef& node) const;
Status EvaluateNode(const NodeDef& node,
@@ -77,7 +74,6 @@ class ConstantFolding : public GraphOptimizer {
GraphDef* output);
// Points to an externally provided device or to owned_device_;
- RewriterConfig::Toggle opt_level_;
DeviceBase* cpu_device_;
std::unique_ptr<DeviceBase> owned_device_;