Home
last modified time | relevance | path

Searched refs:input_lengths (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/st/ops/gpu/
Dtest_ctcloss_op.py46 def construct(self, probs, indices, labels, input_lengths): argument
47 return self.grad(self.network)(probs, indices, labels, input_lengths)
77 input_lengths = Tensor([5, 5, 5], dtype=mstype.int32)
81 ctc_loss = net(probs, labels, input_lengths, indices)
85 grad = GradData(net)(probs, labels, input_lengths, indices)
/third_party/mindspore/tests/st/ops/cpu/
Dtest_ctcloss_op.py46 def construct(self, probs, indices, labels, input_lengths): argument
47 return self.grad(self.network)(probs, indices, labels, input_lengths)
77 input_lengths = Tensor([5, 5, 5], dtype=mstype.int32)
81 ctc_loss = net(probs, labels, input_lengths, indices)
85 grad = GradData(net)(probs, labels, input_lengths, indices)
/third_party/mindspore/mindspore/ops/_grad_experimental/
Dgrad_nn_ops.py30 def bprop(log_probs, targets, input_lengths, target_lengths, out, dout): argument
31 … grad = ctc_loss_grad(dout[1], log_probs, targets, input_lengths, target_lengths, out[0], out[1])
33 return grad, zeros_like(targets), zeros_like(input_lengths), zeros_like(target_lengths)
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.cc1646 Status OperatorInfo::SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths, in SetInputAndOutputTypeLength() argument
1648 if (input_lengths.size() != inputs_shape_.size()) { in SetInputAndOutputTypeLength()
1649 MS_LOG(ERROR) << "Input_lengths: " << input_lengths.size() in SetInputAndOutputTypeLength()
1658 inputs_type_lengths_ = input_lengths; in SetInputAndOutputTypeLength()
1660 operator_cost()->SetInputAndOutputTypeLength(input_lengths, output_lengths); in SetInputAndOutputTypeLength()
Doperator_info.h76 Status SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc35 void OperatorCost::SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths, in SetInputAndOutputTypeLength() argument
37 inputs_type_lengths_ = input_lengths; in SetInputAndOutputTypeLength()
Doperator_costmodel.h66 …void SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths, const std::vector<size_…
/third_party/mindspore/config/
Dop_info.config36 …1, "name": "labels", "param_type": "required"}, {"index": 2, "name": "input_lengths", "param_type"…