Home
last modified time | relevance | path

Searched defs:local_batches (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DPersistentSoftmax.cuh80 int local_batches = batch_size - first_batch; in softmax_warp_forward() local
227 int local_batches = batch_size - first_batch; in softmax_warp_backward() local