Searched refs:end_batch (Results 1 – 1 of 1) sorted by relevance
3651 int start_batch, int end_batch) { in SoftmaxImpl() argument3657 logit_size, end_batch - start_batch); in SoftmaxImpl()3659 end_batch - start_batch); in SoftmaxImpl()3676 int start_batch, int end_batch) in SoftmaxWorkerTask()3683 end_batch(end_batch) {} in SoftmaxWorkerTask()3686 start_batch, end_batch); in Run()3696 int end_batch; member