Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dctcloss_gpu_kernel.h233 int log_prob_size = batch * SOffSet * max_time; in LaunchSecondHalf() local
305 int log_prob_size, int batch, cudaStream_t stream) { in MemManageForCus()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dctcloss_impl.cu162 __global__ void LogBInitKernel(T *log_b, int log_prob_size) { in LogBInitKernel()
281 int log_prob_size = SOffSet * batch * maxtime; in CalculateBwdVar() local
294 int log_prob_size = SOffSet * batch * maxtime; in CalculateFwdVar() local