Searched defs:seq_dim (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | reverse_sequence.cc | 26 void ReverseSequence::Init(const int64_t seq_dim, const int64_t batch_dim) { in Init()
30 void ReverseSequence::set_seq_dim(const int64_t seq_dim) { (void)this->AddAttr(kSeqDim, MakeValue(s… in set_seq_dim()
51 auto seq_dim = GetValue<int64_t>(primitive->GetAttr(kSeqDim)); in ReverseSequenceInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | reverse_sequence_impl.cu | 57 … const int64_t seq_dim, size_t *cur_pos_arr, const size_t *input_shape_ptr, in ReverseSequence()
88 const int64_t seq_dim, size_t *cur_pos_arr, const size_t *input_shape_ptr, in CalReverseSequence()
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_reverse_sequence.py | 27 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reverse_sequence.py | 12 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | searchsorted_cpu_kernel.cc | 62 size_t seq_dim = sequence_shape_.size(); in Launch() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 5623 def __init__(self, seq_dim, batch_dim=0): argument
|