Searched refs:ReverseSequenceGpuFwdKernel (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | reverse_sequence_gpu_kernel.cc | 25 ReverseSequenceGpuFwdKernel, int8_t, int)
29 ReverseSequenceGpuFwdKernel, int8_t, int64_t)
34 ReverseSequenceGpuFwdKernel, int16_t, int)
38 ReverseSequenceGpuFwdKernel, int16_t, int64_t)
43 ReverseSequenceGpuFwdKernel, int, int)
47 ReverseSequenceGpuFwdKernel, int, int64_t)
52 ReverseSequenceGpuFwdKernel, int64_t, int)
56 ReverseSequenceGpuFwdKernel, int64_t, int64_t)
61 ReverseSequenceGpuFwdKernel, half, int)
65 ReverseSequenceGpuFwdKernel, half, int64_t)
[all …]
|
| D | reverse_sequence_gpu_kernel.h | 32 class ReverseSequenceGpuFwdKernel : public GpuKernel {
34 ReverseSequenceGpuFwdKernel() in ReverseSequenceGpuFwdKernel() function
44 ~ReverseSequenceGpuFwdKernel() override = default;
|