Searched refs:num_minibatches (Results 1 – 1 of 1) sorted by relevance
1176 const size_t num_minibatches = [&] { in FastParseExample() local1196 return (serialized.size() * minibatch) / num_minibatches; in FastParseExample()1208 std::vector<std::vector<SparseBuffer>> sparse_buffers(num_minibatches); in FastParseExample()1209 std::vector<std::vector<SparseBuffer>> varlen_dense_buffers(num_minibatches); in FastParseExample()1210 std::vector<std::vector<SparseBuffer>> ragged_buffers(num_minibatches); in FastParseExample()1211 std::vector<Status> status_of_minibatch(num_minibatches); in FastParseExample()1233 ParallelFor(ProcessMiniBatch, num_minibatches, thread_pool); in FastParseExample()