| /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)
|
