Searched defs:log_alpha (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ctcloss_v2_grad_gpu_kernel.cc | 79 auto log_alpha = reinterpret_cast<scalar_t *>(inputs[kIndex6]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | ctcloss_v2_grad_cpu_kernel.cc | 89 … const scalar_t *log_alpha, const NdTensorIterator<kDim3> &log_alpha_it, scalar_t *log_beta, in ComputeGrad()
147 auto log_alpha = static_cast<scalar_t *>(inputs[kIndex6]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | ctcloss_v2_impl.cu | 269 … const S *neg_log_likelihood, const S *log_alpha, S *log_beta, int64_t batch_size, in AlphaBetaInitKernel()
296 const S *neg_log_likelihood, const S *log_alpha, S *log_beta, in AlphaBetaComputeKernel()
363 … const T *target_lengths, const S *neg_log_likelihood, const S *log_alpha, S *log_beta, in CalCTCLossGradV2()
|