Home
last modified time | relevance | path

Searched refs:wrapped_fn (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Dfunction_utils_test.py79 wrapped_fn = functools.partial(fn, test_arg=123)
81 self.assertEqual(('a',), function_utils.fn_args(wrapped_fn))
91 wrapped_fn = functools.partial(fn, 123)
93 self.assertEqual(('a',), function_utils.fn_args(wrapped_fn))
95 self.assertEqual(3, wrapped_fn(3))
96 self.assertEqual(3, wrapped_fn(a=3))
107 wrapped_fn = functools.partial(fn, test_arg2=456)
108 double_wrapped_fn = functools.partial(wrapped_fn, test_arg1=123)
121 wrapped_fn = functools.partial(fn, test_arg2=456)
122 double_wrapped_fn = functools.partial(wrapped_fn, 123)
[all …]
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py415 wrapped_fn = g.wrap_function(fn, signature)
416 self.assertEqual(expected, wrapped_fn(constant_op.constant(5)).numpy())
561 wrapped_fn = wrap_function.wrap_function(
565 self.evaluate(wrapped_fn(v))
Dfunction.py3907 wrapped_fn = self.weakrefself_func__()
3908 if tf_inspect.ismethod(wrapped_fn):
3909 wrapped_fn = six.get_unbound_function(wrapped_fn)
3910 return wrapped_fn(self.weakrefself_target__(), *args, **kwargs)
3935 wrapped_fn = strong_bound_method_wrapper.__wrapped__
3937 if wrapped_fn is strong_bound_method_wrapper.__original_wrapped__:
3940 wrapped_fn = original_function.python_function
3941 if tf_inspect.ismethod(wrapped_fn):
3942 wrapped_fn = six.get_unbound_function(wrapped_fn)
3943 return wrapped_fn(weak_instance(), *args, **kwargs)
[all …]
Ddef_function.py585 def wrapped_fn(*args, **kwds): function
615 weak_wrapped_fn = weakref.ref(wrapped_fn)
619 wrapped_fn))
Dbackprop_test.py564 wrapped_fn = backprop.make_vjp(f, persistent=False)
565 result, vjp = wrapped_fn(constant_op.constant(3.0))
574 wrapped_fn = backprop.make_vjp(f, persistent=True)
575 _, vjp = wrapped_fn(constant_op.constant(3.0))
/external/tensorflow/tensorflow/core/framework/
Ddataset.h383 std::function<void()> wrapped_fn = std::bind( in Params()
386 ctx_runner(std::move(wrapped_fn)); in Params()
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_test.py375 wrapped_fn = wrap_function.wrap_function(my_func, [])
376 wrapped_fn()
/external/tensorflow/tensorflow/python/autograph/g3doc/reference/
Derror_handling.md101 File "tensorflow/python/eager/def_function.py", line 310, in wrapped_fn