Searched refs:def_function (Results 1 – 25 of 64) sorted by relevance
123
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_gradients_test.py | 25 from tensorflow.python.eager import def_function
48 @def_function.function
61 @def_function.function
74 @def_function.function
85 @def_function.function
88 @def_function.function
102 @def_function.function
117 @def_function.function
130 @def_function.function
145 @def_function.function
[all …]
|
| D | def_function_test.py | 25 from tensorflow.python.eager import def_function
53 @def_function.function(
67 @def_function.function
91 @def_function.function
99 @def_function.function
109 @def_function.function
121 @def_function.function
134 @def_function.function
146 @def_function.function
163 @def_function.function
[all …]
|
| D | function_test.py | 34 from tensorflow.python.eager import def_function
102 matmul = def_function.function(math_ops.matmul, autograph=False)
111 add = def_function.function(lambda x, v: x + v1 + v)
195 fn = def_function.function()(layer)
218 @def_function.function
225 @def_function.function
239 @def_function.function()
263 matmul = def_function.function(math_ops.matmul, autograph=False)
265 @def_function.function
275 matmul = def_function.function(math_ops.matmul, autograph=False)
[all …]
|
| D | BUILD | 56 ":def_function",
196 ":def_function",
211 ":def_function",
228 ":def_function",
246 ":def_function",
512 name = "def_function",
513 srcs = ["def_function.py"],
558 ":def_function",
574 ":def_function",
|
| D | function_defun_collection_test.py | 23 from tensorflow.python.eager import def_function
39 function_decorator=def_function.function))
66 function_decorator=def_function.function))
|
| D | def_function_xla_test.py | 20 from tensorflow.python.eager import def_function
33 @def_function.function
|
| D | lift_to_graph_test.py | 21 from tensorflow.python.eager import def_function
35 @def_function.function
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_test.py | 29 from tensorflow.python.eager import def_function
94 root.f = def_function.function(
105 exported.f = def_function.function(
153 root.f = def_function.function(
166 root.f = def_function.function(
196 @def_function.function
213 @def_function.function(
225 @def_function.function
241 f = def_function.function(
244 g = def_function.function(
[all …]
|
| D | save_test.py | 26 from tensorflow.python.eager import def_function
55 @def_function.function(
91 root.f = def_function.function(
103 root.f = def_function.function(
119 root.f = def_function.function(lambda x: 2. * x)
131 @def_function.function
147 root.f = def_function.function(
160 root.f = def_function.function(lambda x: (2. * x, (3. * x, 4. * x)))
170 f=def_function.function(
183 root.f = def_function.function(
[all …]
|
| D | signature_serialization.py | 23 from tensorflow.python.eager import def_function
39 if (isinstance(function, (defun.Function, def_function.Function)) and
105 @def_function.function
210 if isinstance(value, (def_function.Function, defun.ConcreteFunction))
|
| D | BUILD | 276 "//tensorflow/python/eager:def_function",
310 "//tensorflow/python/eager:def_function",
330 "//tensorflow/python/eager:def_function",
392 "//tensorflow/python/eager:def_function",
419 "//tensorflow/python/eager:def_function",
456 "//tensorflow/python/eager:def_function",
469 "//tensorflow/python/eager:def_function",
|
| D | save.py | 29 from tensorflow.python.eager import def_function
149 if isinstance(function, def_function.Function):
174 if isinstance(node, (def_function.Function, defun.ConcreteFunction,
428 resource_initializers.append(def_function.function(
575 elif isinstance(obj, def_function.Function):
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test.py | 26 from tensorflow.python.eager import def_function
58 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
70 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
88 root.f = def_function.function(lambda x: root.v1 * x)
106 root.f = def_function.function(lambda x: 2. * x)
131 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
158 @def_function.function
164 @def_function.function
197 @def_function.function
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants_test.py | 25 from tensorflow.python.eager import def_function
75 root.f = def_function.function(lambda x: 2. * x)
106 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
131 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
164 @def_function.function
170 @def_function.function
201 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
237 @def_function.function
|
| D | auto_control_deps_test.py | 22 from tensorflow.python.eager import def_function
290 @def_function.function
295 @def_function.function
306 @def_function.function
|
| D | config_test.py | 22 from tensorflow.python.eager import def_function
175 @def_function.function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateful_random_ops_test.py | 23 from tensorflow.python.eager import def_function
82 @def_function.function
145 @def_function.function
181 f_acc = def_function.function(f)
207 @def_function.function
233 @def_function.function
274 @def_function.function
|
| D | nccl_ops.py | 23 from tensorflow.python.eager import def_function
234 return def_function.function(_all_reduce)()
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | tracking.py | 20 from tensorflow.python.eager import def_function
104 if isinstance(attribute_value, (def_function.Function,
180 @def_function.function(input_signature=[], autograph=False)
185 @def_function.function(input_signature=[], autograph=False)
|
| D | BUILD | 153 "//tensorflow/python/eager:def_function",
188 "//tensorflow/python/eager:def_function",
250 "//tensorflow/python/eager:def_function",
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | summary_ops_test.py | 33 from tensorflow.python.eager import def_function
73 @def_function.function
103 @def_function.function
133 @def_function.function
215 @def_function.function
238 @def_function.function
329 @def_function.function
369 @def_function.function
373 @def_function.function
393 @def_function.function(input_signature=[
[all …]
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils.py | 21 from tensorflow.python.eager import def_function
62 if isinstance(model.call, def_function.Function):
89 @def_function.function(input_signature=input_signature, autograph=False)
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 22 from tensorflow.python.eager import def_function
65 save_tensor = def_function.function(
67 restore_op = def_function.function(
|
| /external/tensorflow/tensorflow/python/module/ |
| D | module_test.py | 28 from tensorflow.python.eager import def_function
234 self.assertTrue(isinstance(mod.forward, def_function.Function))
235 self.assertTrue(isinstance(mod.forward_ag, def_function.Function))
361 @def_function.function(autograph=False)
366 @def_function.function(autograph=True)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 27 from tensorflow.python.eager import def_function
563 @def_function.function
574 @def_function.function
586 @def_function.function
599 @def_function.function
616 @def_function.function
|
123