Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dctcloss_cpu_kernel.cc41 void InnerSoftMax(const T *inputs_addr, std::vector<std::vector<T>> *softmax_probs, const uint32_t … in InnerSoftMax() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dctcloss_impl.cu303 cudaError_t InnerSoftMax(const T *probs, T *softmax_probs, const int *sequence_length, int max_time… in InnerSoftMax() function