Home
last modified time | relevance | path

Searched defs:KernelRunFunc (Results 1 – 25 of 62) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dupsample_nearest_3d_grad_cpu_kernel.h57 …using KernelRunFunc = std::function<bool(UpsampleNearest3DGradCpuKernelMod *, const std::vector<Ke… variable
Dupsample_nearest_3d_cpu_kernel.h56 …using KernelRunFunc = std::function<bool(UpsampleNearest3DCpuKernelMod *, const std::vector<Kernel… variable
Dupsample_trilinear_3d_cpu_kernel.h81 …using KernelRunFunc = std::function<bool(UpsampleTrilinear3DCpuKernelMod *, const std::vector<Kern… variable
Dupsample_trilinear_3d_grad_cpu_kernel.h81 …using KernelRunFunc = std::function<bool(UpsampleTrilinear3DGradCpuKernelMod *, const std::vector<… variable
Dhsigmoid_grad_cpu_kernel.cc28 using KernelRunFunc = HSigmoidGradCpuKernelMod::KernelRunFunc; typedef
Dspace_to_batch_nd_cpu_kernel.cc30 using KernelRunFunc = SpaceToBatchNDCpuKernelMod::KernelRunFunc; typedef
Dfast_gelu_grad_cpu_kernel.cc28 using KernelRunFunc = FastGeLUGradCpuKernelMod::KernelRunFunc; typedef
Dinplace_index_add_cpu_kernel.cc156 using KernelRunFunc = InplaceIndexAddCpuKernelMod::KernelRunFunc; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Didentity_n_gpu_kernel.cc22 using KernelRunFunc = IdentityNGpuKernelMod::KernelRunFunc; typedef
Dextract_image_patches_gpu_kernel.cc158 using KernelRunFunc = ExtractImagePatchesKernelMod::KernelRunFunc; typedef
Dspacetobatch_gpu_kernel.cc22 using KernelRunFunc = SpaceToBatchGpuKernelMod::KernelRunFunc; typedef
Dspacetodepth_gpu_kernel.cc25 using KernelRunFunc = SpaceToDepthGpuKernelMod::KernelRunFunc; typedef
Dmatrix_diag_part_v3_gpu_kernel.cc26 using KernelRunFunc = MatrixDiagPartV3GpuKernelMod::KernelRunFunc; typedef
Dunsorted_segment_prod_gpu_kernel.cc22 using KernelRunFunc = UnsortedSegmentProdGpuKernelMod::KernelRunFunc; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dmirror_pad_gpu_kernel.cc106 using KernelRunFunc = MirrorPadGpuKernelMod::KernelRunFunc; typedef
Dctcloss_v2_gpu_kernel.cc27 using KernelRunFunc = CTCLossV2GpuKernelMod::KernelRunFunc; typedef
Dctcloss_v2_grad_gpu_kernel.cc24 using KernelRunFunc = CTCLossV2GradGpuKernelMod::KernelRunFunc; typedef
Dadaptive_avg_pool2d_gpu_kernel.cc25 using KernelRunFunc = AdaptiveAvgPool2DKernelMod::KernelRunFunc; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dnms_with_mask_gpu_kernel.cc28 using KernelRunFunc = NMSWithMaskFwdGpuKernelMod::KernelRunFunc; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/convolution/
Dconv2d_filter_grad_gpu_kernel.cc22 using KernelRunFunc = Conv2dFilterGradGpuKernelMod::KernelRunFunc; typedef
Dconv2d_input_grad_gpu_kernel.cc21 using KernelRunFunc = Conv2dInputGradGpuKernelMod::KernelRunFunc; typedef
Dconv2d_forward_gpu_kernel.cc22 using KernelRunFunc = Conv2dFwdGpuKernelMod::KernelRunFunc; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/random/
Drandom_poisson_gpu_kernel.cc31 using KernelRunFunc = RandomPoissonGpuKernelMod::KernelRunFunc; typedef
Dparameterized_truncated_normal_gpu_kernel.cc35 using KernelRunFunc = ParameterizedTruncatedNormalGpuKernelMod::KernelRunFunc; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dmatrix_solve_cpu_kernel.cc29 using KernelRunFunc = MatrixSolveCpuKernelMod::KernelRunFunc; typedef

123