| 24 | namespace toco { |
| 25 | |
| 26 | struct TensorFlowImportFlags { |
| 27 | // If true, control dependencies will be dropped immediately |
| 28 | // during the import of the TensorFlow GraphDef. |
| 29 | bool drop_control_dependency = false; |
| 30 | |
| 31 | // Do not recognize any op and import all ops as |
| 32 | // `TensorFlowUnsupportedOperator`. This is used to populated with the |
| 33 | // `force_select_tf_ops` flag. |
| 34 | bool import_all_ops_as_unsupported = false; |
| 35 | }; |
| 36 | |
| 37 | // Converts TOCO model from TensorFlow GraphDef with given flags. |
| 38 | std::unique_ptr<Model> ImportTensorFlowGraphDef( |
no outgoing calls