HomeSort by relevance Sort by last modified time
    Searched refs:wrap_function (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/python/eager/
wrap_function_test.py 23 from tensorflow.python.eager import wrap_function
48 f_add = wrap_function.wrap_function(
54 # Can call tf.compat.v1.wrap_function again to get a new trace, a new set
56 f_sub = wrap_function.wrap_function(
73 f_wrapped = wrap_function.wrap_function(
84 f_wrapped = wrap_function.wrap_function(f, []
    [all...]
wrap_function.py 105 When we import a GraphDef inside a wrap_function, no Python graph building
219 `tf.compat.v1.wrap_function` to learn more about wrapping V1 functions.
243 add = g.wrap_function(add_v1, [tf.TensorSpec([], tf.int32)])
244 increment_var = g.wrap_function(increment_var_v1,
276 def wrap_function(self, fn, signature, name=None): member in class:WrappedGraph
298 @tf_export(v1=["wrap_function"])
299 def wrap_function(fn, signature, name=None): function
304 created by `fn` will be owned by the object returned by `wrap_function`. The
318 f_add = tf.compat.v1.wrap_function(f, [tf.TensorSpec((), tf.float32), True])
323 # Can call tf.compat.v1.wrap_function again to get a new trace, a new se
    [all...]
  /external/tensorflow/tensorflow/python/training/saving/
functional_saver_test.py 24 from tensorflow.python.eager import wrap_function
56 wrapped = wrap_function.wrap_function(
  /external/tensorflow/tensorflow/python/saved_model/
load_v1_in_v2.py 23 from tensorflow.python.eager import wrap_function
80 """Called from wrap_function to import `meta_graph_def`."""
124 wrapped = wrap_function.wrap_function(
  /external/tensorflow/tensorflow/python/ops/
standard_ops.py 55 from tensorflow.python.eager import wrap_function
  /external/tensorflow/tensorflow/python/kernel_tests/
control_flow_ops_py_test.py 37 from tensorflow.python.eager import wrap_function
    [all...]
variable_scope_test.py 28 from tensorflow.python.eager import wrap_function
52 mode, we wrap the function using wrap_function and then execute the wrapped
63 wrapped = wrap_function.wrap_function(graph_function, [self])
123 # TODO(mihaimaruseac): Not converted to use wrap_function because of
151 # TODO(mihaimaruseac): Not converted to use wrap_function because of
244 # TODO(mihaimaruseac): Not converted to use wrap_function because of
270 # TODO(mihaimaruseac): Not converted to use wrap_function because of
289 # TODO(mihaimaruseac): Not converted to use wrap_function because of
313 # TODO(mihaimaruseac): Not converted to use wrap_function because o
    [all...]

Completed in 357 milliseconds