Home
last modified time | relevance | path

Searched refs:map_defun (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_defun_op_test.py25 from tensorflow.python.data.experimental.ops import map_defun
59 r = map_defun.map_defun(
74 r = map_defun.map_defun(simple_fn, [elems], [dtypes.int32], [(2,)])[0]
87 r = map_defun.map_defun(fn, [elems], [dtypes.int32], [()])[0]
101 r = map_defun.map_defun(fn, [elems], [dtypes.int32], [()])[0]
114 r = map_defun.map_defun(fn, [elems], [dtypes.int32, dtypes.float64], [(2,),
128 result = map_defun.map_defun(fn, [elems], [dtypes.int32], [(2,)])[0]
139 result = map_defun.map_defun(fn, [elems], [dtypes.int32], [(2,)])
154 result = map_defun.map_defun(fn, [elems1, elems2],
171 result = map_defun.map_defun(fn, [elems], [dtypes.int32], [()])
[all …]
DBUILD422 "//tensorflow/python/data/experimental/ops:map_defun",
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dmap_defun_benchmark.py21 from tensorflow.python.data.experimental.ops import map_defun
58 map_defun_op = map_defun.map_defun(defun, [base], [dtypes.int32], [()])
DBUILD127 "//tensorflow/python/data/experimental/ops:map_defun",
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dmap_defun.py26 def map_defun(fn, function
66 return gen_dataset_ops.map_defun(elems, concrete_fn.captured_inputs,
DBUILD198 name = "map_defun",
199 srcs = ["map_defun.py"],
512 ":map_defun",