| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | reverse_sequence.cc | 47 void ReverseSequence::Init(const int64_t seq_dim, const int64_t batch_dim) { in Init()
51 void ReverseSequence::set_seq_dim(const int64_t seq_dim) { (void)this->AddAttr(kSeqDim, api::MakeVa… in set_seq_dim()
78 auto seq_dim = GetValue<int64_t>(seq_dim_ptr); in ReverseSequenceInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | reverse_sequence.cc | 30 static uint32_t CheckSequence(CpuKernelContext &ctx, size_t seq_dim, const Tlen *seq, std::vector<i… in CheckSequence()
42 uint32_t CalReverseSequence(size_t seq_dim, size_t batch_dim, const std::vector<void *> &ioAddrs, in CalReverseSequence()
108 AttrValue *seq_dim = ctx.GetAttr("seq_dim"); in GetInputAndCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | reverse_sequence_impl.cu | 62 … const int64_t seq_dim, size_t *cur_pos_arr, const size_t *input_shape_ptr, in ReverseSequence()
100 … const int64_t seq_dim, size_t *cur_pos_arr, const size_t *input_shape_ptr, in CalReverseSequence()
|
| D | search_sorted_impl.cu | 66 …r_t CalSearchSorted(const size_t size, const S *sequence, const S *values, T *output, int *seq_dim, in CalSearchSorted()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_reverse_sequence.py | 28 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_reverse_sequence.py | 13 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_reverse_sequence.py | 13 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | searchsorted_cpu_kernel.cc | 80 size_t seq_dim = sequence_shape_.size(); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | search_sorted_gpu_kernal.cc | 115 auto seq_dim = GetDeviceAddress<int>(workspace, 0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | fused_infer_attention_score_info.cc | 351 int64_t seq_dim = SizeToLong(dev_matrix_shape_.size()) - dev_matrix_s1_dim_ - 1; in GetSplitIdAndRank() local
646 int64_t seq_dim = SizeToLong(dev_matrix_shape_.size()) - dev_matrix_s1_dim_ - 1; in ComputeReplaceGraphForSplitKVSeq() local
|
| D | prompt_flash_attention_info.cc | 203 int64_t seq_dim = SizeToLong(dev_matrix_shape_.size()) - dev_matrix_s1_dim_ - 1; in GetSplitIdAndRank() local
|
| D | flash_attention_score_info.cc | 186 int64_t seq_dim = SizeToLong(dev_matrix_shape_.size()) - dev_matrix_s1_dim_ - 1; in GetSPRankList() local
928 int64_t seq_dim = SizeToLong(dev_matrix_shape_.size()) - dev_matrix_s1_dim_ - 1; in GetSplitIdAndRank() local
|
| /third_party/mesa3d/include/tensorflow/lite/core/c/ |
| D | builtin_op_data.h | 475 int seq_dim; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_1.cc | 1096 PrimitivePtr MindIR_ReverseSequence_CreatePrimitive(int64_t seq_dim, int64_t batch_dim) { in MindIR_ReverseSequence_CreatePrimitive()
1120 void MindIR_ReverseSequence_SetSeqDim(PrimitivePtr *primitive, int64_t seq_dim) { in MindIR_ReverseSequence_SetSeqDim()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | array_ops.py | 3994 def __init__(self, seq_dim, batch_dim=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | array_func.py | 1572 def reverse_sequence(x, seq_lengths, seq_dim, batch_dim=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 646 def reverse_sequence(x, seq_lengths, seq_dim, batch_dim=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 1879 def reverse_sequence(self, seq_lengths, seq_dim=0, batch_dim=0): argument
|