Searched defs:input_lengths (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_ctcloss_op.py | 46 def construct(self, probs, indices, labels, input_lengths): argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_ctcloss_op.py | 46 def construct(self, probs, indices, labels, input_lengths): argument
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_nn_ops.py | 30 def bprop(log_probs, targets, input_lengths, target_lengths, out, dout): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 1646 Status OperatorInfo::SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths, in SetInputAndOutputTypeLength()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 35 void OperatorCost::SetInputAndOutputTypeLength(const std::vector<size_t> &input_lengths, in SetInputAndOutputTypeLength()
|