| 221 | } // namespace |
| 222 | |
| 223 | Status FreezeSavedModel(const SavedModelBundle& saved_model_bundle, |
| 224 | GraphDef* frozen_graph_def, |
| 225 | std::unordered_set<string>* inputs, |
| 226 | std::unordered_set<string>* outputs) { |
| 227 | GetSignatureDefsInputsAndOutputs(saved_model_bundle, inputs, outputs); |
| 228 | TF_RETURN_IF_ERROR( |
| 229 | FreezeGraphDef(saved_model_bundle, *outputs, frozen_graph_def)); |
| 230 | return Status::OK(); |
| 231 | } |
| 232 | |
| 233 | } // namespace tensorflow |