Searched refs:partial_bins (Results 1 – 1 of 1) sorted by relevance
62 auto partial_bins = partial_bins_t.matrix<T>(); in Compute() local63 partial_bins.setZero(); in Compute()71 partial_bins(worker_id, value) += weights(i); in Compute()74 partial_bins(worker_id, value) += T(1); in Compute()82 output.device(context->eigen_cpu_device()) = partial_bins.sum(reduce_dims); in Compute()