Searched refs:Defun (Results 1 – 25 of 41) sorted by relevance
12
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_test.py | 98 @function.Defun(dtypes.float32, func_name="MyIdentity")
111 @function.Defun(dtypes.float32, func_name="MyIdentity")
126 @function.Defun(
139 @function.Defun(
155 @function.Defun(dtypes.float32, dtypes.float32, func_name="APlus2B")
167 @function.Defun(dtypes.float32, dtypes.float32)
178 @function.Defun(
196 @function.Defun(dtypes.float32, func_name="Duplicate")
213 @function.Defun(dtypes.float32, func_name="XSquarePlusOneFn")
217 @function.Defun(dtypes.float32, dtypes.float32)
[all …]
|
| D | importer_test.py | 1162 @function.Defun(dtype, dtype, dtype, dtype)
1168 @function.Defun(dtype, dtype, grad_func=Grad)
1172 @function.Defun(dtypes.int32)
1177 @function.Defun(dtypes.int32, dtypes.int32)
1180 @function.Defun(dtypes.int32)
1241 @function.Defun()
1251 @function.Defun()
1265 @function.Defun()
|
| D | meta_graph_test.py | 124 @function.Defun(dtypes.int32)
128 @function.Defun(dtypes.int32)
221 @function.Defun(dtypes.float32, dtypes.float32)
225 @function.Defun(dtypes.float32, dtypes.float32)
|
| D | op_def_library_test.py | 469 @function.Defun(dtypes.float32, func_name="MyFn")
485 @function.Defun(dtypes.float32, func_name="MyFn")
488 @function.Defun(dtypes.int32, dtypes.float32, func_name="MyFn2")
491 @function.Defun(dtypes.int32, func_name="MyFn3")
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | functional_ops_test.py | 413 @function.Defun(dtypes.int32, dtypes.int32)
438 @function.Defun(dtypes.int32, dtypes.int32)
461 @function.Defun(dtypes.int32, dtypes.int32)
483 @function.Defun(dtypes.float32, dtypes.float32)
508 @function.Defun(dtypes.float32, dtypes.float32)
533 @function.Defun(dtypes.string)
551 @function.Defun(dtypes.float32, dtypes.float32)
574 @function.Defun(dtypes.float32)
578 @function.Defun(dtypes.float32)
596 @function.Defun(*[dtypes.float32] * 2)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | while_test.py | 42 @function.Defun(dtypes.int32)
48 @function.Defun(dtypes.int32)
62 @function.Defun(dtypes.int32, dtypes.float32)
69 @function.Defun(dtypes.int32, dtypes.float32)
88 @function.Defun(dtypes.int32, dtypes.complex64)
95 @function.Defun(dtypes.int32, dtypes.complex64)
114 @function.Defun(dtypes.int32, dtypes.int32)
121 @function.Defun(dtypes.int32, dtypes.int32)
|
| D | function_test.py | 45 @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)
|
| D | jit_test.py | 60 @function.Defun(func_name=name, noinline=noinline, compiled=True)
135 @function.Defun(compiled=True)
382 @function.Defun(compiled=True)
386 @function.Defun(compiled=True)
390 @function.Defun()
427 @function.Defun(compiled=compiled)
536 @function.Defun(compiled=True)
600 @function.Defun(compiled=True)
630 @function.Defun(compiled=True)
|
| D | reduce_window_test.py | 47 @function.Defun(dtype, dtype)
51 @function.Defun(dtype, dtype)
|
| D | xla_ops_test.py | 213 @function.Defun(dtype, dtype)
242 @function.Defun(dtype, dtype)
263 @function.Defun(dtype, dtype)
267 @function.Defun(dtype, dtype)
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| D | recurrent.py | 336 @function.Defun(*_Dtypes(fwd_sig))
362 @function.Defun(dtypes.int32, dev_t_type, *_Dtypes(fwdloop_sig))
419 @function.Defun(
458 @function.Defun(*_Dtypes(bak_sig))
477 @function.Defun(*_Dtypes(state_if_sig))
483 @function.Defun(*_Dtypes(state_if_sig))
513 @function.Defun(dtypes.int32, dev_t_type, *_Dtypes(bakloop_sig))
552 @function.Defun(*_Dtypes(backward_sig), noinline=noinline)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | map_dataset_serialization_test.py | 97 @function.Defun(dtypes.int64)
110 @function.Defun(dtypes.int64)
113 @function.Defun(dtypes.int32)
|
| D | parallel_map_dataset_serialization_test.py | 109 @function.Defun(dtypes.int64)
123 @function.Defun(dtypes.int64)
126 @function.Defun(dtypes.int32)
|
| D | flat_map_dataset_serialization_test.py | 69 @function.Defun(dtypes.int64)
|
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| D | scaled_softplus.py | 96 @function.Defun(dtype, dtype, dtype, dtype,
105 @function.Defun(dtype, dtype, dtype,
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | jit_test.py | 143 @function.Defun(compiled=True, noinline=True)
159 @function.Defun(compiled=True, noinline=True)
247 @function.Defun(compiled=True, noinline=True)
273 @function.Defun(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops.py | 561 @function.Defun(
623 @function.Defun(*cond_dtypes, func_name="%s_Wrapper" % cond.name)
688 @function.Defun(*body_sig, func_name=cond_name)
695 @function.Defun(*body_sig, func_name=body_name)
|
| D | ctc_ops.py | 765 @function.Defun(*[x.dtype for x in args],
1083 @function.Defun(*loop_dtypes)
1090 @function.Defun(*loop_dtypes)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
| D | ordinal_split_handler.py | 521 @function.Defun(
554 @function.Defun(
594 @function.Defun(
640 @function.Defun(
|
| /external/tensorflow/tensorflow/contrib/batching/python/ops/ |
| D | batch_ops_test.py | 267 @function.Defun(dtypes.int32)
299 @function.Defun(dtypes.int32)
331 @function.Defun(dtypes.int32, dtypes.int32)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | datasets.py | 159 @function.Defun(dtypes.string)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_cluster_test.py | 70 @function.Defun(dtypes.string)
|
| D | iterator_test.py | 606 @function.Defun(dtypes.string)
683 @function.Defun(dtypes.string)
721 @function.Defun(dtypes.uint8)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchFunction.pbtxt | 98 @tf.Defun(tf.float32)
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | make_test_graphs.py | 120 @function.Defun(dtypes.int32, dtypes.int32)
|
12