Home
last modified time | relevance | path

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

1234567891011

/external/tensorflow/tensorflow/python/eager/
Dfunction_gradients_test.py24 from tensorflow.python.eager import def_function
68 @def_function.function
81 @def_function.function
92 @def_function.function
94 y = def_function.function(lambda: math_ops.cos(x))()
106 @def_function.function
113 y = def_function.function(lambda: math_ops.cos(x))()
140 @def_function.function
145 traced_f = def_function.function(f)
150 self.assertAllClose(expected(one), def_function.function(f)(one))
[all …]
Ddef_function_xla_jit_test.py23 from tensorflow.python.eager import def_function
48 @def_function.function(jit_compile=False)
52 @def_function.function(jit_compile=True)
74 func = def_function.function(fn, jit_compile=False)
75 xla_func = def_function.function(fn, jit_compile=True)
84 @def_function.function(jit_compile=True)
97 xla_func = def_function.function(fn, jit_compile=True)
123 @def_function.function(jit_compile=True)
127 @def_function.function(jit_compile=False)
140 xla_func = def_function.function(fn, jit_compile=True)
[all …]
Ddef_function_test.py31 from tensorflow.python.eager import def_function
61 @def_function.function
70 @def_function.function
78 @def_function.function
88 @def_function.function
98 @def_function.function
108 @def_function.function
121 @def_function.function
133 @def_function.function
146 @def_function.function
[all …]
Dfunction_test.py41 from tensorflow.python.eager import def_function
148 matmul = def_function.function(math_ops.matmul)
169 tf_g = def_function.function(g)
177 tf_f = def_function.function(f)
190 add = def_function.function(lambda x, v: x + v1 + v)
198 add = def_function.function(lambda x: x.assign_add(1.0))
221 @def_function.function
251 v = def_function.function(
277 v = def_function.function(
289 add = def_function.function(experimental_implements='MyFunc')(add)
[all …]
Dremote_test.py33 from tensorflow.python.eager import def_function
74 @def_function.function
94 @def_function.function
104 @def_function.function
144 @def_function.function
163 @def_function.function
174 @def_function.function(
181 @def_function.function
215 @def_function.function
238 @def_function.function
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dtf_function_test.py28 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
111 def_function.run_functions_eagerly(run_functions_eagerly)
131 @def_function.function
154 @def_function.function
[all …]
Dtpu_strategy_test.py34 from tensorflow.python.eager import def_function
109 @def_function.function(
123 @def_function.function
127 @def_function.function
176 @def_function.function
189 @def_function.function
216 @def_function.function
236 @def_function.function
263 @def_function.function
274 @def_function.function
[all …]
Dcustom_training_loop_input_test.py31 from tensorflow.python.eager import def_function
86 @def_function.function
110 @def_function.function
152 @def_function.function
171 @def_function.function
208 @def_function.function
231 @def_function.function
255 @def_function.function
286 @def_function.function
311 @def_function.function
[all …]
Dstrategy_gather_test.py32 from tensorflow.python.eager import def_function
81 run = def_function.function(run)
140 run = def_function.function(run)
162 run = def_function.function(run)
190 run = def_function.function(run)
222 def_function.function(run)(t0)
242 run = def_function.function(run)
288 replica_fn = def_function.function(replica_fn)
353 @def_function.function
386 run = def_function.function(run)
[all …]
/external/tensorflow/tensorflow/python/saved_model/
Dload_test.py33 from tensorflow.python.eager import def_function
146 @def_function.function(
163 root.f = def_function.function(
177 root.f = def_function.function(
186 exported.f = def_function.function(
252 @def_function.function(input_signature=[
265 root.f = def_function.function(
278 root.f = def_function.function(
309 @def_function.function
326 @def_function.function(
[all …]
Dsave_test.py33 from tensorflow.python.eager import def_function
93 root.f = def_function.function(
104 @def_function.function
111 root.f = def_function.function(
121 root.f = def_function.function(lambda z: {"out": 2. * z})
155 root.f = def_function.function(annotated_function)
171 @def_function.function(input_signature=[])
176 @def_function.function(input_signature=[])
188 @def_function.function(input_signature=[])
210 root.f = def_function.function(lambda x: 2. * x)
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dtpu_outside_compilation_test.py32 from tensorflow.python.eager import def_function
107 return def_function.function(tpu_fn)
125 return def_function.function(inner_func)
140 @def_function.function
160 @def_function.function
181 @def_function.function
207 @def_function.function
234 @def_function.function
260 @def_function.function
287 @def_function.function
[all …]
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_gpu_test.py26 from tensorflow.python.eager import def_function
67 @def_function.function
87 @def_function.function
109 @def_function.function
131 @def_function.function
152 @def_function.function
175 @def_function.function
193 @def_function.function
214 @def_function.function
235 @def_function.function
[all …]
Dfunctional_ops_test.py21 from tensorflow.python.eager import def_function
50 @def_function.function(
55 @def_function.function(
69 @def_function.function(input_signature=signature)
73 @def_function.function(input_signature=signature)
/external/tensorflow/tensorflow/python/eager/memory_tests/
Dmemory_test.py28 from tensorflow.python.eager import def_function
58 @def_function.function
70 @def_function.function
73 @def_function.function
100 @def_function.function(autograph=False)
103 @def_function.function(autograph=False)
/external/tensorflow/tensorflow/python/compiler/mlir/
Dmlir_test.py22 from tensorflow.python.eager import def_function
50 @def_function.function
63 @def_function.function
67 @def_function.function
80 @def_function.function
/external/tensorflow/tensorflow/compiler/tests/
Drisc_ops_test.py18 from tensorflow.python.eager import def_function
30 @def_function.function(jit_compile=True)
42 @def_function.function(jit_compile=True)
55 @def_function.function(jit_compile=True)
67 @def_function.function(jit_compile=True)
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dgraph_decompose_test.py23 from tensorflow.python.eager import def_function
39 add = def_function.function(gen_composite_ops.my_add_n)
51 biased_dense = def_function.function(gen_composite_ops.my_biased_dense)
59 biased_dense = def_function.function(gen_composite_ops.my_biased_dense)
68 @def_function.function
/external/tensorflow/tensorflow/python/framework/experimental/
Dunified_api_test.py32 from tensorflow.python.framework.experimental import def_function
76 func_output = def_function.function(model)(a, b)
102 func_outputs = def_function.function(model)(a, b)
125 model_fn = def_function.function(model)
155 model_fn = def_function.function(model)
180 func_output = def_function.function(model)(a)
204 func_outputs = def_function.function(model)(a)
225 func_output = def_function.function(model)(a, b)
251 func_outputs = def_function.function(model)(a, b)
274 func_output = def_function.function(model)(a, b)
[all …]
/external/tensorflow/tensorflow/lite/python/
Dlite_v2_test_util.py28 from tensorflow.python.eager import def_function
92 root.f = def_function.function(lambda x: root.v1 * root.v2 * x)
104 @def_function.function
110 @def_function.function
116 @def_function.function
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_test.py34 from tensorflow.python.eager import def_function
128 @def_function.function
142 @def_function.function
175 @def_function.function
189 @def_function.function
222 @def_function.function
264 @def_function.function
360 @def_function.function
446 def_function.function(collective_fn)()
468 def_function.function(collective_fn)()
[all …]
/external/tensorflow/tensorflow/python/keras/distribute/
Dcustom_training_loop_models_test.py32 from tensorflow.python.eager import def_function
75 @def_function.function
100 @def_function.function
138 @def_function.function
163 @def_function.function
195 @def_function.function
238 @def_function.function
292 @def_function.function
304 @def_function.function
309 @def_function.function
[all …]
/external/tensorflow/tensorflow/python/compiler/xla/
Djit_compile_test.py22 from tensorflow.python.eager import def_function
38 xla_func = def_function.function(fn, jit_compile=True)
52 xla_func = def_function.function(fn, jit_compile=True)
76 xla_func = def_function.function(fn, jit_compile=True)
93 xla_func = def_function.function(fn, jit_compile=True)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dplacement_test.py25 from tensorflow.python.eager import def_function
57 @def_function.function
71 @def_function.function
86 @def_function.function
106 @def_function.function
127 @def_function.function
147 @def_function.function
241 @def_function.function
/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps_test.py24 from tensorflow.python.eager import def_function
181 @def_function.function
292 @def_function.function
305 @def_function.function
308 @def_function.function
323 @def_function.function
326 @def_function.function
346 @def_function.function
349 @def_function.function
372 @def_function.function
[all …]

1234567891011