Home
last modified time | relevance | path

Searched refs:def_function (Results 1 – 25 of 344) sorted by relevance

12345678910>>...14

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction_gradients_test.py20 from tensorflow.python.eager import def_function
65 @def_function.function
78 @def_function.function
89 @def_function.function
91 y = def_function.function(lambda: math_ops.cos(x))()
103 @def_function.function
110 y = def_function.function(lambda: math_ops.cos(x))()
137 @def_function.function
142 traced_f = def_function.function(f)
147 self.assertAllClose(expected(one), def_function.function(f)(one))
[all …]
H A Ddef_function_xla_jit_test.py22 from tensorflow.python.eager import def_function
52 @def_function.function(jit_compile=False)
56 @def_function.function(jit_compile=True)
78 func = def_function.function(fn, jit_compile=False)
79 xla_func = def_function.function(fn, jit_compile=True)
88 @def_function.function(jit_compile=True)
101 xla_func = def_function.function(fn, jit_compile=True)
127 @def_function.function(jit_compile=True)
131 @def_function.function(jit_compile=False)
148 xla_func = def_function.function(fn, jit_compile=True)
[all …]
H A Ddef_function_test.py29 from tensorflow.python.eager import def_function
60 @def_function.function
69 @def_function.function
77 @def_function.function
87 @def_function.function
97 @def_function.function
107 @def_function.function
120 @def_function.function
132 @def_function.function
145 @def_function.function
[all …]
H A Dfunction_test.py39 from tensorflow.python.eager import def_function
147 matmul = def_function.function(math_ops.matmul)
156 def_function.function(1)
172 tf_g = def_function.function(g)
180 tf_f = def_function.function(f)
193 add = def_function.function(lambda x, v: x + v1 + v)
201 add = def_function.function(lambda x: x.assign_add(1.0))
209 @def_function.function
213 @def_function.function
238 @def_function.function
[all …]
H A Dremote_test.py29 from tensorflow.python.eager import def_function
71 @def_function.function
91 @def_function.function
101 @def_function.function
169 @def_function.function
185 @def_function.function
196 @def_function.function(
203 @def_function.function
222 @def_function.function
255 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dload_test.py37 from tensorflow.python.eager import def_function
156 @def_function.function(
173 root.f = def_function.function(
187 root.f = def_function.function(
196 exported.f = def_function.function(
262 @def_function.function(input_signature=[
275 root.f = def_function.function(
288 root.f = def_function.function(
326 @def_function.function
343 @def_function.function(
[all …]
H A Dsave_test.py32 from tensorflow.python.eager import def_function
94 root.f = def_function.function(
105 @def_function.function
112 root.f = def_function.function(
122 root.f = def_function.function(lambda z: {"out": 2. * z})
156 root.f = def_function.function(annotated_function)
172 @def_function.function(input_signature=[])
177 @def_function.function(input_signature=[])
189 @def_function.function(input_signature=[])
212 root.f = def_function.function(lambda x: 2. * x)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_strategy_test.py30 from tensorflow.python.eager import def_function
103 @def_function.function(
126 @def_function.function(
167 @def_function.function(
195 @def_function.function(
209 @def_function.function
213 @def_function.function
263 @def_function.function
285 @def_function.function
302 @def_function.function
[all …]
H A Dcustom_training_loop_input_test.py27 from tensorflow.python.eager import def_function
84 @def_function.function
108 @def_function.function
150 @def_function.function
169 @def_function.function
206 @def_function.function
229 @def_function.function
253 @def_function.function
284 @def_function.function
309 @def_function.function
[all …]
H A Dtf_function_test.py24 from tensorflow.python.eager import def_function
44 def_function.run_functions_eagerly(False)
55 def_function.run_functions_eagerly(run_functions_eagerly)
64 @def_function.function
81 def_function.run_functions_eagerly(run_functions_eagerly)
89 @def_function.function
93 @def_function.function
114 def_function.run_functions_eagerly(run_functions_eagerly)
134 @def_function.function
157 @def_function.function
[all …]
H A Dstrategy_gather_test.py28 from tensorflow.python.eager import def_function
79 run = def_function.function(run)
138 run = def_function.function(run)
160 run = def_function.function(run)
188 run = def_function.function(run)
215 def_function.function(run)(t0)
235 run = def_function.function(run)
281 replica_fn = def_function.function(replica_fn)
346 @def_function.function
379 run = def_function.function(run)
[all …]
H A Dvars_test.py32 from tensorflow.python.eager import def_function
93 update_fn = def_function.function(update_fn)
138 update_fn = def_function.function(update_fn)
188 update_fn = def_function.function(update_fn)
231 @def_function.function
249 @def_function.function
275 read_var_fn = def_function.function(v.read_value)
300 read_var_fn = def_function.function(v.read_value)
326 @def_function.function
352 @def_function.function
[all …]
H A Dstrategy_common_test.py28 from tensorflow.python.eager import def_function
52 @def_function.function
56 @def_function.function
68 @def_function.function
101 @def_function.function
112 @def_function.function
123 @def_function.function
135 @def_function.function
164 @def_function.function
176 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_outside_compilation_test.py29 from tensorflow.python.eager import def_function
105 return def_function.function(tpu_fn)
123 return def_function.function(inner_func)
138 @def_function.function
158 @def_function.function
179 @def_function.function
205 @def_function.function
232 @def_function.function
258 @def_function.function
285 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dplacement_test.py21 from tensorflow.python.eager import def_function
54 @def_function.function
67 @def_function.function
82 @def_function.function
102 @def_function.function
121 @def_function.function
138 @def_function.function
151 @def_function.function
155 @def_function.function
193 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcollective_ops_gpu_test.py22 from tensorflow.python.eager import def_function
63 @def_function.function
83 @def_function.function
105 @def_function.function
127 @def_function.function
148 @def_function.function
171 @def_function.function
189 @def_function.function
210 @def_function.function
231 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dlight_outside_compilation_test.py19 from tensorflow.python.eager import def_function
44 @def_function.function(jit_compile=True)
62 @def_function.function(jit_compile=True)
74 @def_function.function(jit_compile=True)
95 @def_function.function(jit_compile=True)
116 @def_function.function(jit_compile=True)
132 @def_function.function(jit_compile=True)
152 @def_function.function(jit_compile=True)
174 @def_function.function(jit_compile=True)
190 @def_function.function(jit_compile=True)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A Dcollective_ops_test.py29 from tensorflow.python.eager import def_function
137 @def_function.function
152 @def_function.function
192 @def_function.function
207 @def_function.function
239 @def_function.function
269 @def_function.function
316 @def_function.function
451 @def_function.function
502 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dlite_v2_test_util.py24 from tensorflow.python.eager import def_function
93 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
105 @def_function.function
121 @def_function.function
127 @def_function.function
133 @def_function.function
151 @def_function.function
155 @def_function.function
159 @def_function.function
178 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A Dtpu_embedding_v2_invalid_input_test.py20 from tensorflow.python.eager import def_function
59 @def_function.function
67 @def_function.function
86 @def_function.function
115 @def_function.function
129 @def_function.function
154 @def_function.function
168 @def_function.function
188 @def_function.function
212 @def_function.function
H A Dtpu_embedding_v2_enqueue_mode_test.py21 from tensorflow.python.eager import def_function
50 @def_function.function
57 @def_function.function
92 @def_function.function
96 @def_function.function
122 @def_function.function
150 @def_function.function
161 @def_function.function
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dcluster_coordinator_test.py40 from tensorflow.python.eager import def_function
216 @def_function.function()
516 @def_function.function
529 @def_function.function
546 @def_function.function
576 @def_function.function
621 @def_function.function
715 @def_function.function
719 @def_function.function
741 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dunified_api_test.py28 from tensorflow.python.framework.experimental import def_function
73 func_output = def_function.function(model)(a, b)
99 func_outputs = def_function.function(model)(a, b)
122 model_fn = def_function.function(model)
152 model_fn = def_function.function(model)
177 func_output = def_function.function(model)(a)
201 func_outputs = def_function.function(model)(a)
222 func_output = def_function.function(model)(a, b)
248 func_outputs = def_function.function(model)(a, b)
271 func_output = def_function.function(model)(a, b)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dauto_control_deps_test.py20 from tensorflow.python.eager import def_function
77 @def_function.function
92 @def_function.function
117 @def_function.function
270 @def_function.function
381 @def_function.function
394 @def_function.function
397 @def_function.function
412 @def_function.function
415 @def_function.function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/
H A Dstateful_random_ops_test.py24 from tensorflow.python.eager import def_function
116 @def_function.function
143 @def_function.function
153 @def_function.function
171 @def_function.function
201 @def_function.function
216 @def_function.function
262 @def_function.function
282 @def_function.function
339 @def_function.function
[all …]

12345678910>>...14