Searched refs:batch_reduce_fn (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 97 def batch_reduce_fn(state, value): function
100 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)
141 def batch_reduce_fn(state, value): function
149 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)
290 def batch_reduce_fn(state, value): function
300 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)
346 def batch_reduce_fn(state, value): function
355 reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)
|