Home
last modified time | relevance | path

Searched refs:finalize_fn (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dbatching.py79 def finalize_fn(state): function
88 finalize_fn)
100 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)
118 def finalize_fn(state): function
127 finalize_fn)
149 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)
266 def finalize_fn(state): function
270 max_reducer = grouping.Reducer(max_init_fn, max_reduce_fn, finalize_fn)
300 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)
328 def finalize_fn(state): function
[all …]
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
Dreduce_dataset_test.py49 def finalize_fn(state): function
52 sum_reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py181 def finalize_fn(s1, s2): function
184 reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_generator_test.py363 def finalize_fn(_): function
372 finalize_fn).take(2)
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py491 def finalize_fn(iterator_id_t): function
515 finalize_fn)