Searched refs:finalize_fn (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | batching.py | 79 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/ |
D | reduce_dataset_test.py | 49 def finalize_fn(state): function 52 sum_reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
|
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
D | group_by_reducer_test.py | 181 def finalize_fn(s1, s2): function 184 reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
|
/external/tensorflow/tensorflow/python/data/kernel_tests/ |
D | from_generator_test.py | 363 def finalize_fn(_): function 372 finalize_fn).take(2)
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | dataset_ops.py | 491 def finalize_fn(iterator_id_t): function 515 finalize_fn)
|