Searched defs:seq_lengths (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_reverse_sequence.py | 32 def construct(self, x, seq_lengths): argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reverse_sequence.py | 18 def construct(self, x, seq_lengths): argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | reverse_sequence.cc | 50 …auto seq_lengths = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kS… in ReverseSequenceInfer() local
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | dataset.py | 160 def construct(self, seq_lengths): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 5631 def infer_shape(self, x, seq_lengths): argument
5640 def infer_dtype(self, x, seq_lengths): argument
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 1051 def bprop(x, seq_lengths, out, dout): argument
|