diff options
Diffstat (limited to 'tensorflow/core/grappler/optimizers/constant_folding.h')
-rw-r--r-- | tensorflow/core/grappler/optimizers/constant_folding.h | 6 |
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_; |