def _load_and_run_model(self,
distribution,
saved_dir,
predict_dataset,
output_name='output_1'):
return test_base.load_and_run_with_saved_model_api(
distribution, saved_dir, predict_dataset, output_name)
def _load_and_run_model(self, distribution, saved_dir, predict_dataset,
output_name, experimental_run_tf_function):
return test_base.load_and_run_with_saved_model_api(distribution, saved_dir,
predict_dataset,
output_name)