Import a SavedModel into a TF 1.x-style graph and run `signature_key`.
(save_dir, inputs)
| 236 | |
| 237 | |
| 238 | def _import_and_infer(save_dir, inputs): |
| 239 | """Import a SavedModel into a TF 1.x-style graph and run `signature_key`.""" |
| 240 | graph = ops.Graph() |
| 241 | with graph.as_default(), session_lib.Session() as session: |
| 242 | model = loader.load(session, [tag_constants.SERVING], save_dir) |
| 243 | signature = model.signature_def[ |
| 244 | signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY] |
| 245 | assert set(inputs.keys()) == set(signature.inputs.keys()) |
| 246 | feed_dict = {} |
| 247 | for arg_name in inputs.keys(): |
| 248 | feed_dict[graph.get_tensor_by_name(signature.inputs[arg_name].name)] = ( |
| 249 | inputs[arg_name]) |
| 250 | output_dict = {} |
| 251 | for output_name, output_tensor_info in signature.outputs.items(): |
| 252 | output_dict[output_name] = graph.get_tensor_by_name( |
| 253 | output_tensor_info.name) |
| 254 | return session.run(output_dict, feed_dict=feed_dict) |
| 255 | |
| 256 | |
| 257 | class ModelSaveTest(keras_parameterized.TestCase): |
no test coverage detected