| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | custom_training_loop_test.py | 71 def add_loss_step(defun): argument
85 if defun:
92 def batch_norm_step(defun): argument
108 if defun:
115 def add_metric_step(defun): argument
134 if defun:
151 eager_result = train_step(defun=False)
152 fn_result = train_step(defun=True)
156 def test_training_arg_propagation(self, defun): argument
164 if defun:
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | map_defun_op_test.py | 53 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
68 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
81 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
95 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
108 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
122 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
134 @function.defun(input_signature=[tensor_spec.TensorSpec([2], dtypes.int32)])
145 @function.defun(input_signature=[
165 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
178 @function.defun(input_signature=[tensor_spec.TensorSpec([5], dtypes.int64)])
[all …]
|
| D | get_single_element_test.py | 93 @function.defun
120 @function.defun
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test.py | 96 def _apply(self, defun=False, execution_mode=None): argument
99 if defun:
109 self._apply(defun=False)
112 self._apply(defun=False, execution_mode=context.ASYNC)
115 self._apply(defun=True)
118 self._apply(defun=True, execution_mode=context.ASYNC)
303 def _benchmark_eager_apply(self, label, device_and_format, defun=False, argument
308 if defun:
330 defun=False)
336 defun=False,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 305 matmul = function.defun(math_ops.matmul)
327 model = function.defun(model)
337 @function.defun
403 f = function.defun(f)
415 f = function.defun(f)
429 f = function.defun(f)
463 foo = function.defun(foo)
485 @function.defun
502 @function.defun
516 @function.defun
[all …]
|
| D | cond_test.py | 47 @function.defun
67 @function.defun
227 @function.defun
|
| /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 | 300 @function.defun
323 @function.defun
326 @function.defun
345 @function.defun
348 @function.defun
351 @function.defun
512 @function.defun
570 @function.defun
573 @function.defun
616 @function.defun
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | map_defun_benchmark.py | 48 @function.defun(input_signature=[tensor_spec.TensorSpec([], dtypes.int32)])
49 def defun(x): function
58 map_defun_op = map_defun.map_defun(defun, [base], [dtypes.int32], [()])
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 107 @function.defun()
123 @function.defun()
134 @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/saved_model/ |
| D | signature_serialization.py | 24 from tensorflow.python.eager import function as defun unknown
44 if (isinstance(function, (defun.Function, def_function.Function)) and
47 if not isinstance(function, defun.ConcreteFunction):
258 if isinstance(value, (def_function.Function, defun.ConcreteFunction))
284 assert isinstance(func, defun.ConcreteFunction)
|
| D | function_serialization.py | 22 from tensorflow.python.eager import function as defun unknown
158 inner_concrete = defun.ConcreteFunction(concrete_function.graph)
168 fn = defun.ConcreteFunction(
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_defun_collection_test.py | 36 dict(testcase_name='Defun', function_decorator=function.defun),
63 dict(testcase_name='Defun', function_decorator=function.defun),
86 @function.defun
|
| D | function_test.py | 398 @function.defun
409 @function.defun(experimental_relax_shapes=True)
595 @function.defun(experimental_relax_shapes=True)
673 @function.defun
725 function_decorator=function.defun),
753 @function.defun
1028 defined = function.defun(f)
1068 @function.defun
1085 defined = function.defun(f)
1251 @function.defun
[all …]
|
| D | forwardprop.py | 192 _jvp_relaxed_shapes = function.defun(
194 _jvp_exact_shapes = 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/keras/saving/saved_model/ |
| D | serialized_attributes.py | 22 from tensorflow.python.eager import function as defun unknown
194 (defun.Function, def_function.Function))):
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 50 @function.defun(autograph=False) # Pure graph code.
57 @function.defun(autograph=False) # Pure graph code.
69 @function.defun(
110 @function.defun(
119 @function.defun(
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_strategy_util.py | 87 @function.defun
188 @function.defun
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | tracking.py | 27 from tensorflow.python.eager import function as defun unknown
124 defun.ConcreteFunction)):
|
| /external/libffi/doc/ |
| D | libffi.texi | 142 @defun ffi_status ffi_prep_cif (ffi_cif *@var{cif}, ffi_abi @var{abi}, unsigned int @var{nargs}, ff…
162 @end defun
168 @defun ffi_status ffi_prep_cif_var (ffi_cif *@var{cif}, ffi_abi @var{abi}, unsigned int @var{nfixed…
186 @end defun
197 @defun void ffi_call (ffi_cif *@var{cif}, void *@var{fn}, void *@var{rvalue}, void **@var{avalues})
233 @end defun
484 @defun ffi_status ffi_get_struct_offsets (ffi_abi abi, ffi_type *struct_type, size_t *offsets)
500 @end defun
804 @defun void *ffi_closure_alloc (size_t @var{size}, void **@var{code})
810 @end defun
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy_test.py | 1058 for defun in defuns:
1065 defun.get_concrete_function, args=[mock_model] + inputs))
1076 @function.defun
1086 @function.defun
1090 @function.defun
1101 @function.defun
1105 @function.defun
1116 @function.defun
1120 @function.defun
1135 @function.defun
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 93 @function.defun(autograph=autograph)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph_test.py | 201 @function.defun
223 @function.defun
|
| D | auto_control_deps_test.py | 716 @function.defun
746 @function.defun
758 optimizer.apply_gradients = function.defun(optimizer.apply_gradients)
776 @function.defun
792 @function.defun
|