/external/tensorflow/tensorflow/python/util/ |
tf_contextlib.py | 22 from tensorflow.python.util import tf_decorator 26 """A tf_decorator-aware wrapper for `contextlib.contextmanager`. 36 return tf_decorator.make_decorator(target, context_manager, 'contextmanager')
|
tf_decorator_test.py | 15 """Unit tests for tf_decorator.""" 26 from tensorflow.python.util import tf_decorator 33 return tf_decorator.TFDecorator(decorator_name, target, decorator_doc) 52 return tf_decorator.make_decorator(target, wrapper) 60 return tf_decorator.make_decorator(target, wrapper) 68 return tf_decorator.make_decorator(target, wrapper) 114 tf_decorator.TFDecorator('', test_function).decorated_target) 118 tf_decorator.TFDecorator('decorator name', 123 tf_decorator.TFDecorator('', test_function, 133 tf_decorator.TFDecorator('', test_function, '' [all...] |
tf_inspect.py | 26 from tensorflow.python.util import tf_decorator 120 decorators, target = tf_decorator.unwrap(obj) 248 decorators, target = tf_decorator.unwrap(obj) 306 unwrapped_object = tf_decorator.unwrap(object)[1] 335 return _inspect.getsource(tf_decorator.unwrap(object)[1]) 340 return _inspect.getsourcefile(tf_decorator.unwrap(object)[1]) 345 return _inspect.getsourcelines(tf_decorator.unwrap(object)[1]) 350 return _inspect.isbuiltin(tf_decorator.unwrap(object)[1]) 355 return _inspect.isclass(tf_decorator.unwrap(object)[1]) 360 return _inspect.isfunction(tf_decorator.unwrap(object)[1] [all...] |
function_utils.py | 26 from tensorflow.python.util import tf_decorator 31 _, fn = tf_decorator.unwrap(fn) 88 _, func = tf_decorator.unwrap(func) 103 _, func = tf_decorator.unwrap(func)
|
tf_contextlib_test.py | 24 from tensorflow.python.util import tf_decorator 79 decorators, target = tf_decorator.unwrap(test_params_and_defaults) 81 self.assertTrue(isinstance(decorators[0], tf_decorator.TFDecorator)) 83 self.assertFalse(isinstance(target, tf_decorator.TFDecorator))
|
tf_should_use.py | 28 from tensorflow.python.util import tf_decorator 194 return tf_decorator.make_decorator( 228 return tf_decorator.make_decorator(
|
dispatch.py | 35 from tensorflow.python.util import tf_decorator 191 return tf_decorator.make_decorator(target, wrapper)
|
tf_export.py | 48 from tensorflow.python.util import tf_decorator 116 _, undecorated_symbol = tf_decorator.unwrap(symbol) 329 _, undecorated_f = tf_decorator.unwrap(f) 333 _, undecorated_func = tf_decorator.unwrap(func) 395 return tf_decorator.make_decorator(f, wrapper, decorator_argspec=f_argspec)
|
tf_inspect_test.py | 27 from tensorflow.python.util import tf_decorator 34 return tf_decorator.TFDecorator(decorator_name, target, decorator_doc) 85 decorator = tf_decorator.TFDecorator('', test_undecorated_function, '', 96 inner_decorator = tf_decorator.TFDecorator('', test_undecorated_function, 98 outer_decorator = tf_decorator.TFDecorator('', inner_decorator) 107 inner_decorator = tf_decorator.TFDecorator('', test_undecorated_function, 109 outer_decorator = tf_decorator.TFDecorator('', inner_decorator, '', 249 decorator = tf_decorator.TFDecorator('', test_undecorated_function, '', 323 decorator = tf_decorator.TFDecorator('', test_undecorated_function, '', 337 inner_decorator = tf_decorator.TFDecorator('', test_undecorated_function [all...] |
deprecation.py | 29 from tensorflow.python.util import tf_decorator 116 This decorator wraps a function using `tf_decorator.make_decorator` 130 return tf_decorator.make_decorator(wrapped_function, wrapper_func) 162 For functions, it returns a function wrapped by `tf_decorator.make_decorator`. 239 return tf_decorator.make_decorator( 325 return tf_decorator.make_decorator( 511 return tf_decorator.make_decorator(func, new_func, 'deprecated', doc) 578 return tf_decorator.make_decorator(func, new_func, 'deprecated', doc)
|
tf_export_test.py | 25 from tensorflow.python.util import tf_decorator 196 return tf_decorator.make_decorator(func, wrapper)
|
/external/tensorflow/tensorflow/tools/compatibility/update/ |
generate_v2_renames_map.py | 31 from tensorflow.python.util import tf_decorator 81 _, attr = tf_decorator.unwrap(child[1]) 136 _, attr = tf_decorator.unwrap(child[1])
|
generate_v2_reorders_map.py | 29 from tensorflow.python.util import tf_decorator 83 _, attr = tf_decorator.unwrap(child[1])
|
/external/tensorflow/tensorflow/contrib/framework/python/ops/ |
arg_scope.py | 67 from tensorflow.python.util import tf_decorator 186 return tf_decorator.make_decorator(func, func_with_args)
|
/external/tensorflow/tensorflow/python/ops/ragged/ |
ragged_operators.py | 24 from tensorflow.python.util import tf_decorator 29 return tf_decorator.make_decorator(operator, lambda y, x: operator(x, y))
|
ragged_dispatch.py | 44 from tensorflow.python.util import tf_decorator 472 _, undecorated_op = tf_decorator.unwrap(op) 516 return (tf_export.get_v1_names(tf_decorator.unwrap(op)[1]) or 519 return tf_export.get_v2_names(tf_decorator.unwrap(op)[1])
|
/external/tensorflow/tensorflow/python/platform/ |
flags.py | 28 from tensorflow.python.util import tf_decorator 60 return tf_decorator.make_decorator(original_function, wrapper)
|
googletest.py | 36 from tensorflow.python.util import tf_decorator 183 _, obj = tf_decorator.unwrap(obj)
|
/external/tensorflow/tensorflow/python/module/ |
module.py | 27 from tensorflow.python.util import tf_decorator 261 return tf_decorator.make_decorator(method, method_with_name_scope)
|
/external/tensorflow/tensorflow/python/tools/api/generator/ |
output_init_files_test.py | 29 from tensorflow.python.util import tf_decorator 60 _, attr = tf_decorator.unwrap(attr)
|
/external/tensorflow/tensorflow/python/keras/utils/ |
metrics_utils.py | 38 from tensorflow.python.util import tf_decorator 76 return tf_decorator.make_decorator(update_state_fn, decorated) 126 return tf_decorator.make_decorator(result_fn, decorated)
|
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/ |
export_mnist_cnn.py | 32 from tensorflow.python.util import tf_decorator 105 # tf.function uses tf_inspect, we can use tf_decorator to wrap it with 110 tf_decorator.make_decorator(call_fn, wrapped, decorator_argspec=argspec))
|
/external/tensorflow/tensorflow/python/framework/ |
auto_control_deps.py | 30 from tensorflow.python.util import tf_decorator 365 return tf_decorator.make_decorator(f, wrapper)
|
/external/tensorflow/tensorflow/tools/api/lib/ |
python_object_to_proto_visitor.py | 26 from tensorflow.python.util import tf_decorator 185 _, member_obj = tf_decorator.unwrap(member_obj)
|
/external/tensorflow/tensorflow/python/ops/ |
custom_gradient.py | 33 from tensorflow.python.util import tf_decorator 164 return tf_decorator.make_decorator(f, decorated)
|