Searched refs:MS_REG_GPU_KERNEL (Results 1 – 18 of 18) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/debug/ |
| D | print_gpu_kernel.cc | 21 MS_REG_GPU_KERNEL(Print, PrintGpuKernel)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/rl/ |
| D | buffer_append_gpu_kernel.h | 54 MS_REG_GPU_KERNEL(BufferAppend, BufferAppendKernel)
|
| D | buffer_get_gpu_kernel.h | 53 MS_REG_GPU_KERNEL(BufferGetItem, BufferGetKernel)
|
| D | buffer_sample_gpu_kernel.h | 61 MS_REG_GPU_KERNEL(BufferSample, BufferSampleKernel)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/ |
| D | dataset_init_kernel.h | 55 MS_REG_GPU_KERNEL(InitDataSetQueue, DatasetInitKernel)
|
| D | dataset_iterator_kernel.h | 58 MS_REG_GPU_KERNEL(GetNext, DatasetIteratorKernel)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/ |
| D | trt_kernel.h | 54 MS_REG_GPU_KERNEL(TrtNode, TrtKernel)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | minmax_update_perchannel_gpu_kernel.cc | 101 MS_REG_GPU_KERNEL(MinMaxUpdatePerChannel, MinMaxUpdatePerChannelGpuKernel)
|
| D | minmax_update_perlayer_gpu_kernel.cc | 95 MS_REG_GPU_KERNEL(MinMaxUpdatePerLayer, MinMaxUpdatePerLayerGpuKernel)
|
| D | fake_learned_scale_quant_perchannel_gpu_kernel.cc | 128 MS_REG_GPU_KERNEL(FakeLearnedScaleQuantPerChannel, FakeLearnedScaleQuantPerChannelGpuKernel)
|
| D | fake_learned_scale_quant_perlayer_gpu_kernel.cc | 119 MS_REG_GPU_KERNEL(FakeLearnedScaleQuantPerLayer, FakeLearnedScaleQuantPerLayerGpuKernel)
|
| D | fake_learned_scale_quant_perlayer_grad_gpu_kernel.cc | 127 MS_REG_GPU_KERNEL(FakeLearnedScaleQuantPerLayerGrad, FakeLearnedScaleQuantPerLayerGradGpuKernel)
|
| D | fake_quant_perlayer_grad_gpu_kernel.cc | 135 MS_REG_GPU_KERNEL(FakeQuantPerLayerGrad, FakeQuantPerLayerGradGpuKernel)
|
| D | fake_quant_perchannel_grad_gpu_kernel.cc | 138 MS_REG_GPU_KERNEL(FakeQuantPerChannelGrad, FakeQuantPerChannelGradGpuKernel)
|
| D | fake_quant_perlayer_gpu_kernel.cc | 148 MS_REG_GPU_KERNEL(FakeQuantPerLayer, FakeQuantPerLayerGpuKernel)
|
| D | fake_learned_scale_quant_perchannel_grad_gpu_kernel.cc | 132 MS_REG_GPU_KERNEL(FakeLearnedScaleQuantPerChannelGrad, FakeLearnedScaleQuantPerChannelGradGpuKernel)
|
| D | fake_quant_perchannel_gpu_kernel.cc | 156 MS_REG_GPU_KERNEL(FakeQuantPerChannel, FakeQuantPerChannelGpuKernel)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 87 #define MS_REG_GPU_KERNEL(OPNAME, OPCLASS) \ macro
|