Searched refs:batch_fn (Results 1 – 1 of 1) sorted by relevance
407 batch_fn, argument418 batch_ops = batch_fn([t.tensor for t in labeled_tensors], scope)