diff options
| author |  Yu-Cheng Ling <ycling@google.com> | 2018-09-05 06:01:51 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-05 06:06:23 -0700 |
| commit | 858f4672e25825bc5e091a79fd4234f1968a278d (patch) | |
| tree | c2f41b48ff253d104e37b4a8ef52614f239669f2 /tensorflow/contrib/lite/toco/import_tensorflow.h | |
| parent | f15e8613aa42f7f2b1439c652a465438553df219 (diff) | |
Minimum change for generating Eager ops with Toco.
PiperOrigin-RevId: 211621189
Diffstat (limited to 'tensorflow/contrib/lite/toco/import_tensorflow.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/import_tensorflow.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/import_tensorflow.h b/tensorflow/contrib/lite/toco/import_tensorflow.h index 2177872334..7db23f2d44 100644 --- a/tensorflow/contrib/lite/toco/import_tensorflow.h +++ b/tensorflow/contrib/lite/toco/import_tensorflow.h @@ -27,6 +27,11 @@ struct TensorFlowImportFlags { // If true, control dependencies will be dropped immediately // during the import of the TensorFlow GraphDef. bool drop_control_dependency = false; + + // Do not recognize any op and import all ops as + // `TensorFlowUnsupportedOperator`. This is used to populated with the + // `force_eager_ops` flag. + bool import_all_ops_as_unsupported = false; }; std::unique_ptr<Model> ImportTensorFlowGraphDef( |