Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DPersistentSoftmax.cuh323 int batches_per_block = warps_per_block * batches_per_warp; in dispatch_softmax_forward() local
373 int batches_per_block = warps_per_block * batches_per_warp; in dispatch_softmax_backward() local