Searched refs:cur_pos_arr (Results 1 – 3 of 3) sorted by relevance
| /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() argument
66 IdxToPos(idx, cur_pos_arr, cur_thread_idx, input_shape_cum_ptr, shape_size); in ReverseSequence()
67 …cur_slice = cur_pos_arr[cur_thread_idx + batch_dim]; // all accesses to cur_pos_arr have to be ad… in ReverseSequence()
72 if (cur_pos_arr[cur_thread_idx + seq_dim] > (cur_slice_seq_len - 1)) { // check if within range in ReverseSequence()
78 cur_pos_arr[cur_thread_idx + seq_dim] = in ReverseSequence()
79 …(cur_slice_seq_len - 1) - cur_pos_arr[cur_thread_idx + seq_dim]; // adjust positio… in ReverseSequence()
80 …new_idx = PosToIdx(cur_pos_arr, cur_thread_idx, input_shape_cum_ptr, shape_size); // get the upda… in ReverseSequence()
88 const int64_t seq_dim, size_t *cur_pos_arr, const size_t *input_shape_ptr, in CalReverseSequence() argument
92 …size, input, seq_len, batch_dim, seq_dim, cur_pos_arr, input_shape_ptr, input_shape_cum_ptr, shape… in CalReverseSequence()
97 … const int64_t batch_dim, const int64_t seq_dim, size_t *cur_pos_arr,
[all …]
|
| D | reverse_sequence_impl.cuh | 24 const int64_t seq_dim, size_t *cur_pos_arr, const size_t *input_shape_ptr,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | reverse_sequence_gpu_kernel.h | 57 size_t *cur_pos_arr = GetDeviceAddress<size_t>(workspace, 2); in Launch() local
63 …CalReverseSequence(input_size_, input, seq_len, batch_dim_, seq_dim_, cur_pos_arr, input_shape_ptr, in Launch()
|