| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_ctclossv2_op.py | 30 def construct(self, input_matrix, target, input_lengths, target_lengths): argument
41 def construct(self, probs, indices, labels, input_lengths): argument
109 def compare_to_numpy(method, input_matrix, target, input_lengths, target_lengths): argument
|
| D | test_ctcloss_op.py | 46 def construct(self, probs, indices, labels, input_lengths): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_ctclossv2_op.py | 31 def construct(self, input_matrix, target, input_lengths, target_lengths): argument
42 def construct(self, probs, indices, labels, input_lengths): argument
110 def compare_to_numpy(method, input_matrix, target, input_lengths, target_lengths): argument
|
| D | test_ctcloss_op.py | 46 def construct(self, probs, indices, labels, input_lengths): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_ctcloss_v2.py | 29 def construct(self, input_matrix, target, input_lengths, target_lengths): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | ctcloss_v2_impl.cu | 193 __global__ void LogBetaKernel(const S *log_probs, const T *targets, const T *input_lengths, const T… in LogBetaKernel()
268 __global__ void AlphaBetaInitKernel(const T *targets, const T *input_lengths, const T *target_lengt… in AlphaBetaInitKernel()
295 __global__ void AlphaBetaComputeKernel(const T *targets, const T *input_lengths, const T *target_le… in AlphaBetaComputeKernel()
331 __global__ void GradOutKernel(const S *grad_out, const S *log_probs, const T *input_lengths, in GradOutKernel()
362 …t CalCTCLossGradV2(const S *grad_out, const S *log_probs, const T *targets, const T *input_lengths, in CalCTCLossGradV2()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | ctcloss_v2_grad_gpu_kernel.cc | 76 auto input_lengths = reinterpret_cast<target_t *>(inputs[kIndex3]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | ctcloss_v2_grad_cpu_kernel.cc | 144 auto input_lengths = static_cast<target_t *>(inputs[kIndex3]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/loss/ |
| D | loss.py | 2667 def construct(self, log_probs, targets, input_lengths, target_lengths): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 35 void OperatorCost::SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths, in SetInputAndOutputTypeLength()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 2587 Status OperatorInfo::SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths, in SetInputAndOutputTypeLength()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | nn_func.py | 5017 def ctc_loss(log_probs, targets, input_lengths, target_lengths, blank=0, reduction="mean", zero_inf… argument
|