(self, saved_model_dir)
| 56 | return preds |
| 57 | |
| 58 | def load_model(self, saved_model_dir): |
| 59 | saved_model_loaded = tf.saved_model.load(saved_model_dir, tags=[tag_constants.SERVING]) |
| 60 | wrapper_fp32 = saved_model_loaded.signatures['serving_default'] |
| 61 | |
| 62 | self.loaded_model_fn = wrapper_fp32 |
| 63 | |
| 64 | class ModelOptimizer(): |
| 65 | def __init__(self, input_saved_model_dir, calibration_data=None): |
no test coverage detected