Searched defs:batch_map_func (Results 1 – 3 of 3) sorted by relevance
36 py::function batch_map_func, in BatchNode()
99 Builder &SetBatchMapFunc(py::function batch_map_func) { in SetBatchMapFunc()
54 py::function batch_size_func, py::function batch_map_func, PadInfo pad_map) in BatchOp()