Searched refs:batch_size_func_ (Results 1 – 4 of 4) sorted by relevance
44 batch_size_func_(batch_size_func), in BatchNode()60 col_order_, batch_size_func_, batch_map_func_, pad_map_); in Copy()106 … in_col_names_, out_col_names_, batch_size_func_, batch_map_func_, pad_map_); in Build()126 if (batch_size_func_) { in GetDatasetSize()
98 const py::function &BatchSizeFunc() const { return batch_size_func_; } in BatchSizeFunc()124 py::function batch_size_func_; variable
61 batch_size_func_(batch_size_func), in BatchOp()329 if (batch_size_func_) { in GetBatchSize()350 py::object size = batch_size_func_(info); in InvokeBatchSizeFunc()571 if (batch_size_func_) { in GetTreeBatchSize()
282 py::function batch_size_func_; // Function pointer of batch size function variable