Home
last modified time | relevance | path

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

12345678910>>...14

/external/tensorflow/tensorflow/python/eager/
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 …]
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 …]
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 …]
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 …]
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 …]
/external/tensorflow/tensorflow/python/distribute/
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 …]
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 …]
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 …]
/external/tensorflow/tensorflow/compiler/tf2xla/
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)
/external/tensorflow/tensorflow/python/tpu/
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 …]
/external/tensorflow/tensorflow/python/saved_model/
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 …]
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 …]
/external/tensorflow/tensorflow/python/tpu/tests/
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
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
/external/tensorflow/tensorflow/lite/python/
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 …]
/external/tensorflow/tensorflow/python/ops/
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 …]
Dfunctional_ops_test.py17 from tensorflow.python.eager import def_function
46 @def_function.function(
51 @def_function.function(
65 @def_function.function(input_signature=signature)
69 @def_function.function(input_signature=signature)
Dscript_ops_test.py17 from tensorflow.python.eager import def_function
47 @def_function.function
51 @def_function.function
72 @def_function.function
76 @def_function.function
Dembedding_ops_test.py19 from tensorflow.python.eager import def_function
37 @def_function.function(input_signature=[])
41 @def_function.function(input_signature=[])
77 @def_function.function(input_signature=[])
81 @def_function.function(input_signature=[])
/external/tensorflow/tensorflow/python/data/kernel_tests/
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 …]
/external/tensorflow/tensorflow/python/eager/memory_tests/
Dmemory_test.py24 from tensorflow.python.eager import def_function
54 @def_function.function
66 @def_function.function
69 @def_function.function
96 @def_function.function(autograph=False)
99 @def_function.function(autograph=False)
/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/python/compiler/mlir/
Dmlir_test.py18 from tensorflow.python.eager import def_function
50 @def_function.function(
121 @def_function.function
134 @def_function.function
138 @def_function.function
151 @def_function.function
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dgraph_decompose_test.py19 from tensorflow.python.eager import def_function
35 add = def_function.function(gen_composite_ops.my_add_n)
47 biased_dense = def_function.function(gen_composite_ops.my_biased_dense)
55 biased_dense = def_function.function(gen_composite_ops.my_biased_dense)
64 @def_function.function
/external/tensorflow/tensorflow/core/function/
Druntime_client_test.py23 from tensorflow.python.eager import def_function
110 @def_function.function
126 @def_function.function
147 @def_function.function
167 @def_function.function
187 @def_function.function

12345678910>>...14