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)
|