Searched defs:softmax_probs (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | ctcloss_impl.cu | 30 __global__ void CalculateFwdVarKernel(T *log_alpha_b, int *label_value_with_blank, T *softmax_probs, in CalculateFwdVarKernel() 89 __global__ void CalculateBwdVarKernel(T *log_beta_b, int *label_value_with_blank, T *softmax_probs, in CalculateBwdVarKernel() 169 __global__ void CTCLossKernel(T *log_alpha_b, T *log_beta_b, T *softmax_probs, int *label_value_wit… in CTCLossKernel() 220 __global__ void InnerSoftMaxKernel(const T *probs, T *softmax_probs, const int *sequence_length, in… in InnerSoftMaxKernel() 277 cudaError_t CalculateBwdVar(T *log_beta_b, int *label_value_with_blank, T *softmax_probs, const int… in CalculateBwdVar() 290 cudaError_t CalculateFwdVar(T *log_alpha_b, int *label_value_with_blank, T *softmax_probs, const in… in CalculateFwdVar() 303 cudaError_t InnerSoftMax(const T *probs, T *softmax_probs, const int *sequence_length, int max_time… in InnerSoftMax() 430 cudaError_t CTCLoss(T *log_alpha_b, T *log_beta_b, T *softmax_probs, int *label_value_with_blank, i… in CTCLoss()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ctcloss_gpu_kernel.h | 344 T *softmax_probs; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | ctcloss_cpu_kernel.cc | 41 void InnerSoftMax(const T *inputs_addr, std::vector<std::vector<T>> *softmax_probs, const uint32_t … in InnerSoftMax()
|