Home
last modified time | relevance | path

Searched defs:sequence_length (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/tests/st/networks/models/bert/src/
Dcluener_evaluation.py30 def process(model, text, sequence_length): argument
60 def submit(model, path, sequence_length): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dctcloss_impl.cu32const int *sequence_length, bool ctc_merge_repeated, int batch, int SOffSet, in CalculateFwdVarKernel()
91const int *sequence_length, bool ctc_merge_repeated, int batch, int SOffSet, in CalculateBwdVarKernel()
171 int SOffSet, int maxtime, int numclass, const int *sequence_length, in CTCLossKernel()
221 __global__ void InnerSoftMaxKernel(const T *probs, T *softmax_probs, const int *sequence_length, in… in InnerSoftMaxKernel()
278 …ateBwdVar(T *log_beta_b, int *label_value_with_blank, T *softmax_probs, const int *sequence_length, in CalculateBwdVar()
289 …teFwdVar(T *log_alpha_b, int *label_value_with_blank, T *softmax_probs, const int *sequence_length, in CalculateFwdVar()
300 void InnerSoftMax(const T *probs, T *softmax_probs, const int *sequence_length, int max_time, int b… in InnerSoftMax()
399 __global__ void CalculateMaxSequenceKernel(const int *sequence_length, int *max_labels_length, int … in CalculateMaxSequenceKernel()
408 void CalculateMaxSequence(const int *sequence_length, int *max_labels_length, int batch, cudaStream… in CalculateMaxSequence()
421 …int maxtime, int numclass, const int *sequence_length, int *label_squence_length, int *cum_labels_… in CTCLoss()
/third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/
Dtest_ctc_greedy_decoder.py30 def construct(self, inputs, sequence_length): argument
Dtest_ctc_loss.py30 def construct(self, inputs, labels_indices, labels_values, sequence_length): argument
/third_party/mindspore/mindspore/core/ops/
Dctcloss.cc39 …auto sequence_length = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_ar… in CheckCTCLossInputs() local
/third_party/boost/libs/detail/test/
Dbinary_search_test.cpp43 const std::size_t sequence_length = 1000; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dctcloss_gpu_kernel.h297 int *sequence_length; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dctcloss_cpu_kernel.cc40 …x(const T *inputs_addr, std::vector<std::vector<T>> *softmax_probs, const uint32_t sequence_length, in InnerSoftMax()
/third_party/cJSON/
DcJSON.c655 unsigned char sequence_length = 0; in utf16_literal_to_utf8() local
826 unsigned char sequence_length = 2; in parse_string() local
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_nn_ops.py1202 def bprop(inputs, labels_indices, labels_values, sequence_length, out, dout): argument
/third_party/mindspore/tests/ut/python/ops/
Dtest_ops.py915 def construct(self, inputs, sequence_length): argument