Home
last modified time | relevance | path

Searched refs:Defun (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py99 @function.Defun(dtypes.float32, func_name="MyIdentity")
112 @function.Defun(dtypes.float32, func_name="MyIdentity")
127 @function.Defun(
140 @function.Defun(
156 @function.Defun(dtypes.float32, dtypes.float32, func_name="APlus2B")
168 @function.Defun(dtypes.float32, dtypes.float32)
179 @function.Defun(
197 @function.Defun(dtypes.float32, func_name="Duplicate")
214 @function.Defun(dtypes.float32, func_name="XSquarePlusOneFn")
218 @function.Defun(dtypes.float32, dtypes.float32)
[all …]
Dgraph_util_test.py399 @function.Defun(dtypes.float32)
416 @function.Defun(dtypes.float32)
420 @function.Defun(dtypes.float32)
441 @function.Defun(dtypes.float32)
Dimporter_test.py1149 @function.Defun(dtype, dtype, dtype, dtype)
1155 @function.Defun(dtype, dtype, grad_func=Grad)
1159 @function.Defun(dtypes.int32)
1164 @function.Defun(dtypes.int32, dtypes.int32)
1167 @function.Defun(dtypes.int32)
1230 @function.Defun()
1240 @function.Defun()
1256 @function.Defun()
Dmeta_graph_test.py125 @function.Defun(dtypes.int32)
129 @function.Defun(dtypes.int32)
245 @function.Defun(dtypes.float32, dtypes.float32)
249 @function.Defun(dtypes.float32, dtypes.float32)
Dop_def_library_test.py398 @function.Defun(dtypes.float32, func_name="MyFn")
440 @function.Defun(dtypes.float32, func_name="MyFn")
443 @function.Defun(dtypes.int32, dtypes.float32, func_name="MyFn2")
446 @function.Defun(dtypes.int32, func_name="MyFn3")
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py414 @function.Defun(dtypes.int32, dtypes.int32)
439 @function.Defun(dtypes.int32, dtypes.int32)
462 @function.Defun(dtypes.int32, dtypes.int32)
484 @function.Defun(dtypes.float32, dtypes.float32)
509 @function.Defun(dtypes.float32, dtypes.float32)
534 @function.Defun(dtypes.string)
552 @function.Defun(dtypes.float32, dtypes.float32)
575 @function.Defun(dtypes.float32)
579 @function.Defun(dtypes.float32)
597 @function.Defun(*[dtypes.float32] * 2)
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dfunction_test.py45 @function.Defun(dtypes.float32, dtypes.float32)
71 @function.Defun(dtypes.float32, dtypes.float32)
95 @function.Defun(dtypes.float32, dtypes.float32)
110 @function.Defun(dtypes.float32, dtypes.int32, dtypes.int32)
131 @function.Defun(dtypes.float32, noinline=True)
135 @function.Defun(dtypes.float32, dtypes.float32)
Dwhile_test.py43 @function.Defun(dtypes.int32)
49 @function.Defun(dtypes.int32)
63 @function.Defun(dtypes.int32, dtypes.float32)
70 @function.Defun(dtypes.int32, dtypes.float32)
89 @function.Defun(dtypes.int32, dtypes.complex64)
96 @function.Defun(dtypes.int32, dtypes.complex64)
115 @function.Defun(dtypes.int32, dtypes.int32)
122 @function.Defun(dtypes.int32, dtypes.int32)
Djit_test.py61 @function.Defun(func_name=name, noinline=noinline, compiled=True)
142 @function.Defun(compiled=True)
388 @function.Defun(compiled=True)
392 @function.Defun(compiled=True)
396 @function.Defun()
433 @function.Defun(compiled=compiled)
524 @function.Defun(compiled=True)
588 @function.Defun(compiled=True)
618 @function.Defun(compiled=True)
Dreduce_window_test.py47 @function.Defun(dtype, dtype)
51 @function.Defun(dtype, dtype)
Dsort_ops_test.py100 @function.Defun(key_type, key_type)
119 @function.Defun(key_type, key_type)
154 @function.Defun(key_type, key_type, value_type_1, value_type_1,
194 @function.Defun(key_type_1, key_type_1, key_type_2, key_type_2,
226 @function.Defun(key_type, key_type, value_type_1, value_type_1,
Dxla_ops_test.py343 @function.Defun(dtype, dtype)
372 @function.Defun(dtype, dtype)
466 @function.Defun(dtype, dtype)
470 @function.Defun(dtype, dtype)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dmap_dataset_serialization_test.py103 @function.Defun(dtypes.int64)
117 @function.Defun(dtypes.int64)
120 @function.Defun(dtypes.int32)
Dparallel_map_dataset_serialization_test.py114 @function.Defun(dtypes.int64)
129 @function.Defun(dtypes.int64)
132 @function.Defun(dtypes.int32)
Dflat_map_dataset_serialization_test.py77 @function.Defun(dtypes.int64)
/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops_test.py36 @function.Defun(dtypes.float32)
40 @function.Defun(dtypes.float32)
Dbatch_ops_test.py354 @function.Defun(dtypes.int32)
388 @function.Defun(dtypes.int32)
422 @function.Defun(dtypes.int32, dtypes.int32)
448 @function.Defun(dtypes.int32)
Dfunctional_ops.py905 @function.Defun(*_GetInputDtypes(func), func_name="%s_Wrapper" % func.name)
968 @function.Defun(*cond_dtypes, func_name="%s_Wrapper" % cond.name)
1033 @function.Defun(*body_sig, func_name=cond_name)
1040 @function.Defun(*body_sig, func_name=body_name)
/external/tensorflow/tensorflow/python/compiler/xla/
Djit_test.py156 @function.Defun(compiled=True, noinline=True)
173 @function.Defun(compiled=True, noinline=True)
265 @function.Defun(compiled=True, noinline=True)
292 @function.Defun(
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Ddefun_export.py28 @function.Defun(tf.float32, tf.float32)
/external/tensorflow/tensorflow/python/tpu/
Ddatasets.py164 @function.Defun(dtypes.string)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_cluster_test.py73 @function.Defun(dtypes.string)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BatchFunction.pbtxt106 @tf.Defun(tf.float32)
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2_test.py593 @framework_function.Defun(dtypes.int64)
597 @framework_function.Defun(dtypes.int64)
601 @framework_function.Defun(dtypes.int64)
/external/tensorflow/tensorflow/compiler/aot/tests/
Dmake_test_graphs.py126 @function.Defun(dtypes.int32, dtypes.int32)

12