Searched defs:batch_dim (Results 1 – 5 of 5) 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()
31 void ReverseSequence::set_batch_dim(const int64_t batch_dim) { (void)this->AddAttr(kBatchDim, MakeV… in set_batch_dim()
52 auto batch_dim = GetValue<int64_t>(primitive->GetAttr(kBatchDim)); in ReverseSequenceInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | reverse_sequence_impl.cu | 56 … void ReverseSequence(const size_t size, const T *input, const S *seq_len, const int64_t batch_dim, in ReverseSequence()
87 …id CalReverseSequence(const size_t size, const T *input, const S *seq_len, const int64_t batch_dim, 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/ops/operations/ |
| D | array_ops.py | 5623 def __init__(self, seq_dim, batch_dim=0): argument
|