Searched refs:batch_init_fn (Results 1 – 1 of 1) sorted by relevance
92 def batch_init_fn(_): function100 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)131 def batch_init_fn(_): function149 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)285 def batch_init_fn(_): function300 batch_reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)336 def batch_init_fn(_): function355 reducer = grouping.Reducer(batch_init_fn, batch_reduce_fn, finalize_fn)