Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dsoftmax_impl.cu335 int batch_element_count = (i >= local_batches) ? 0 : element_count; in SoftMaxWarpForward() local
349 int batch_element_count = (i >= local_batches) ? 0 : element_count; in SoftMaxWarpForward() local
380 int batch_element_count = (i >= local_batches) ? 0 : element_count; in SoftMaxWarpForward() local