diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-22 05:15:18 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-22 05:19:06 -0700 |
| commit | ca552d54ac67be8837aeabdb43269846d9df4eb5 (patch) | |
| tree | 11d592685766ab64187b520d91d7dfa2b6f231fc /tensorflow/core/protobuf | |
| parent | e317152dad1aa66bc493abc046a60dbbf650de92 (diff) | |
Add PinToHostOptimizer to grappler: force small ops to happen on CPU (instead of
GPU). This avoids many unnecessary CPU<->GPU memcpy and syncs.
PiperOrigin-RevId: 214108484
Diffstat (limited to 'tensorflow/core/protobuf')
| -rw-r--r-- | tensorflow/core/protobuf/rewriter_config.proto | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/protobuf/rewriter_config.proto b/tensorflow/core/protobuf/rewriter_config.proto index 07f984ceea..0e780eacc9 100644 --- a/tensorflow/core/protobuf/rewriter_config.proto +++ b/tensorflow/core/protobuf/rewriter_config.proto @@ -75,6 +75,8 @@ message RewriterConfig { // Try to allocate some independent Op outputs contiguously in order to // merge or eliminate downstream Ops (off by default). Toggle scoped_allocator_optimization = 15; + // Force small ops onto the CPU (default is OFF). + Toggle pin_to_host_optimization = 18; // Controls how many times we run the optimizers in meta optimizer (default // is once). |