| /external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/ |
| D | resnet50_test.py | 76 def _apply(self, defun=False, execution_mode=None): argument
79 if defun:
88 self._apply(defun=False)
91 self._apply(defun=False, execution_mode=tfe.ASYNC)
94 self._apply(defun=True)
97 self._apply(defun=True, execution_mode=tfe.ASYNC)
218 def _benchmark_eager_apply(self, label, device_and_format, defun=False, argument
223 if defun:
244 defun=False)
248 'eager_apply_async', device_and_data_format(), defun=False,
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_defun_op_test.py | 46 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
60 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
72 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
85 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
97 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
110 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
121 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
131 @function.defun(input_signature=[
150 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
162 @function.defun(input_signature=[tensor_spec.TensorSpec([5], dtypes.int64)])
[all …]
|
| D | get_single_element_test.py | 86 @function.defun
112 @function.defun
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/ |
| D | densenet_test.py | 220 def _benchmark_eager_apply(self, label, device_and_format, defun=False, argument
230 if defun:
232 model.call = tfe.defun(model.call)
252 defun=False)
256 'eager_apply_async', device_and_data_format(), defun=False,
261 device_and_data_format(), defun=True)
267 defun=False, argument
281 if defun:
282 model.call = tfe.defun(model.call)
283 apply_grads = tfe.defun(apply_gradients)
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | revnet_test.py | 125 y, _ = tfe.defun(self.model.call)(self.x, training=False)
132 compute_gradients = tfe.defun(self.model.compute_gradients)
233 defun=False, argument
239 if defun:
241 model.call = tfe.defun(model.call)
261 "eager_apply_sync", device_and_data_format(), defun=False)
267 defun=False,
272 "eager_apply_with_defun", device_and_data_format(), defun=True)
278 defun=False, argument
287 if defun:
[all …]
|
| D | main.py | 55 model.call = tfe.defun(model.call)
56 model.compute_gradients = tfe.defun(model.compute_gradients)
57 model.get_moving_stats = tfe.defun(model.get_moving_stats)
58 model.restore_moving_stats = tfe.defun(model.restore_moving_stats)
60 apply_gradients = tfe.defun(apply_gradients)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 305 matmul = function.defun(math_ops.matmul)
327 model = function.defun(model)
337 @function.defun
352 f = function.defun(f)
364 f = function.defun(f)
378 f = function.defun(f)
412 foo = function.defun(foo)
434 @function.defun
451 @function.defun
465 @function.defun
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 93 @function.defun
124 @function.defun
135 @function.defun
156 @function.defun
253 @function.defun
308 @function.defun
504 defined = function.defun(f)
668 @function.defun
681 [function.defun(f).get_concrete_function()
693 @function.defun
[all …]
|
| D | function_defun_collection_test.py | 36 dict(testcase_name='Defun', function_decorator=function.defun),
63 dict(testcase_name='Defun', function_decorator=function.defun),
87 @function.defun
|
| D | benchmarks_test.py | 370 f = function.defun(math_ops.matmul)
375 inner = function.defun(math_ops.matmul)
377 @function.defun
392 f = function.defun(math_ops.matmul)
647 defined = function.defun(func)
658 defined = function.defun(func)
670 defined = function.defun(
682 defined = function.defun(
912 @function.defun
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/l2hmc/ |
| D | README.md | 34 ## Boosting Performance with `tfe.defun`
36 To improve performance, we could wrap certain functions with the decorator `@tfe.defun`.
40 @tfe.defun
46 We could also explicitly wrap the desired function with `tfe.defun`:
49 apply_transition = tfe.defun(apply_transition)
|
| D | l2hmc_test.py | 195 self._benchmark_eager(defun=True)
197 def _benchmark_eager(self, defun=False): argument
210 step_fn = tfe.defun(step) if defun else step
235 "_defun" if defun else "", sample_size),
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | signature_serialization.py | 24 from tensorflow.python.eager import function as defun unknown
39 if (isinstance(function, (defun.Function, def_function.Function)) and
42 if not isinstance(function, defun.ConcreteFunction):
210 if isinstance(value, (def_function.Function, defun.ConcreteFunction))
236 assert isinstance(func, defun.ConcreteFunction)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_util_v2_test.py | 55 @function.defun
56 def defun(): function
59 defun()
|
| D | cond_v2_test.py | 213 @function.defun
236 @function.defun
239 @function.defun
258 @function.defun
261 @function.defun
264 @function.defun
425 @function.defun
483 @function.defun
486 @function.defun
529 @function.defun
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_defun_benchmark.py | 55 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
56 def defun(x): function
65 map_defun_op = map_defun.map_defun(defun, [base], [dtypes.int32], [()])
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 106 @function.defun()
122 @function.defun()
133 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.string)])
166 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.string)])
185 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.string)])
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 46 @function.defun(autograph=False) # Pure graph code.
53 @function.defun(autograph=False) # Pure graph code.
65 @function.defun(
105 @function.defun(
114 @function.defun(
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | auto_control_deps_test.py | 203 @function.defun
233 @function.defun
245 optimizer.apply_gradients = function.defun(optimizer.apply_gradients)
263 @function.defun
279 @function.defun
|
| D | function_def_to_graph_test.py | 197 @function.defun
200 @function.defun
206 @function.defun
231 @function.defun
|
| /external/python/cpython2/Modules/_ctypes/libffi/doc/ |
| D | libffi.texi | 129 @defun ffi_status ffi_prep_cif (ffi_cif *@var{cif}, ffi_abi @var{abi}, unsigned int @var{nargs}, ff…
149 @end defun
155 @defun ffi_status ffi_prep_cif_var (ffi_cif *@var{cif}, ffi_abi var{abi}, unsigned int @var{nfixeda…
173 @end defun
180 @defun void ffi_call (ffi_cif *@var{cif}, void *@var{fn}, void *@var{rvalue}, void **@var{avalues})
199 @end defun
467 @defun void *ffi_closure_alloc (size_t @var{size}, void **@var{code})
473 @end defun
476 @defun void ffi_closure_free (void *@var{writable})
479 @end defun
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | reduce_test.py | 141 @function.defun
160 @function.defun
183 @function.defun
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | tracking.py | 21 from tensorflow.python.eager import function as defun unknown
105 defun.ConcreteFunction)):
|
| /external/libffi/doc/ |
| D | libffi.texi | 129 @defun ffi_status ffi_prep_cif (ffi_cif *@var{cif}, ffi_abi @var{abi}, unsigned int @var{nargs}, ff…
149 @end defun
155 @defun ffi_status ffi_prep_cif_var (ffi_cif *@var{cif}, ffi_abi var{abi}, unsigned int @var{nfixeda…
173 @end defun
180 @defun void ffi_call (ffi_cif *@var{cif}, void *@var{fn}, void *@var{rvalue}, void **@var{avalues})
199 @end defun
612 @defun void *ffi_closure_alloc (size_t @var{size}, void **@var{code})
618 @end defun
621 @defun void ffi_closure_free (void *@var{writable})
624 @end defun
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | tfe.py | 145 defun = _function_lib.defun variable
|