def _test_tf_saved_model_format_no_traces(f, test_or_class, *args, **kwargs): with testing_utils.saved_model_format_scope('tf', save_traces=False): f(test_or_class, *args, **kwargs)
def _test_tf_saved_model_format(f, test_or_class, *args, **kwargs): with testing_utils.saved_model_format_scope('tf'): f(test_or_class, *args, **kwargs)