diff options
| author |  Yao Zhang <yaozhang@google.com> | 2017-09-21 14:52:28 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-21 14:55:13 -0700 |
| commit | 57498a86c11dfc98dda84dc7318a3c84c85c6791 (patch) | |
| tree | af27c5e5c9a264ca86ad25faeb32948c89c618df /tensorflow/core/debug | |
| parent | 847aa2fec14e7cdde140a3e5fdb0c3229caf9426 (diff) | |
Fold fetch nodes.
PiperOrigin-RevId: 169604180
Diffstat (limited to 'tensorflow/core/debug')
| -rw-r--r-- | tensorflow/core/debug/debug_gateway_test.cc | 4 | ||||
| -rw-r--r-- | tensorflow/core/debug/grpc_session_debug_test.cc | 4 |
2 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/debug/debug_gateway_test.cc b/tensorflow/core/debug/debug_gateway_test.cc index 9a74a4bb4c..3903040e4d 100644 --- a/tensorflow/core/debug/debug_gateway_test.cc +++ b/tensorflow/core/debug/debug_gateway_test.cc @@ -26,6 +26,7 @@ limitations under the License. #include "tensorflow/core/lib/core/notification.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/lib/core/threadpool.h" +#include "tensorflow/core/protobuf/rewriter_config.pb.h" namespace tensorflow { namespace { @@ -36,6 +37,9 @@ std::unique_ptr<DirectSession> CreateSession() { options.config.mutable_graph_options() ->mutable_optimizer_options() ->set_opt_level(OptimizerOptions_Level_L0); + options.config.mutable_graph_options() + ->mutable_rewrite_options() + ->set_constant_folding(RewriterConfig::OFF); return std::unique_ptr<DirectSession>( dynamic_cast<DirectSession*>(NewSession(options))); diff --git a/tensorflow/core/debug/grpc_session_debug_test.cc b/tensorflow/core/debug/grpc_session_debug_test.cc index d6f35fe24c..642a2a4c07 100644 --- a/tensorflow/core/debug/grpc_session_debug_test.cc +++ b/tensorflow/core/debug/grpc_session_debug_test.cc @@ -35,6 +35,7 @@ limitations under the License. #include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/test.h" #include "tensorflow/core/protobuf/debug.pb.h" +#include "tensorflow/core/protobuf/rewriter_config.pb.h" #include "tensorflow/core/public/session.h" #include "tensorflow/core/util/port.h" @@ -85,6 +86,9 @@ SessionOptions Options(const string& target, int placement_period) { options.config.mutable_graph_options() ->mutable_optimizer_options() ->set_opt_level(OptimizerOptions::L0); + options.config.mutable_graph_options() + ->mutable_rewrite_options() + ->set_constant_folding(RewriterConfig::OFF); return options; } |