Searched refs:CUfunction (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/stub/runtime/ |
| D | cuda.h | 69 typedef struct CUfunc_st *CUfunction; typedef
75 CUresult cuModuleGetFunction(CUfunction *hfunc, CUmodule hmod, const char *name);
76 CUresult cuLaunchKernel(CUfunction f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gr…
|
| D | cuda.cc | 25 CUresult cuModuleGetFunction(CUfunction *hfunc, CUmodule hmod, const char *name) { return CUDA_SUCC… in cuModuleGetFunction()
27 CUresult cuLaunchKernel(CUfunction f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gr… in cuLaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_mod.h | 29 CUfunction func_addr_;
32 GpuKernelMeta(CUfunction funcAddr, CUmodule module, const std::vector<uint32_t> &thread_info) in GpuKernelMeta()
49 CUfunction *func);
|
| D | akg_gpu_kernel_mod.cc | 37 vector<uint32_t> *thread_info, CUfunction *func) { in GetFunction()
113 CUfunction kernel_addr; in Launch()
|
| /third_party/ffmpeg/libavfilter/ |
| D | vf_yadif_cuda.c | 40 CUfunction cu_func_uchar;
41 CUfunction cu_func_uchar2;
42 CUfunction cu_func_ushort;
43 CUfunction cu_func_ushort2;
53 static CUresult call_kernel(AVFilterContext *ctx, CUfunction func, in call_kernel()
135 CUfunction func; in filter()
|
| D | vf_scale_cuda.c | 99 CUfunction cu_func_uchar;
100 CUfunction cu_func_uchar2;
101 CUfunction cu_func_uchar4;
102 CUfunction cu_func_ushort;
103 CUfunction cu_func_ushort2;
104 CUfunction cu_func_ushort4;
396 static int call_resize_kernel(AVFilterContext *ctx, CUfunction func, int channels, in call_resize_kernel()
|
| D | vf_thumbnail_cuda.c | 65 CUfunction cu_func_uchar;
66 CUfunction cu_func_uchar2;
67 CUfunction cu_func_ushort;
68 CUfunction cu_func_ushort2;
157 static int thumbnail_kernel(AVFilterContext *ctx, CUfunction func, int channels, in thumbnail_kernel()
|
| D | vf_overlay_cuda.c | 71 CUfunction cu_func;
|
| /third_party/gstreamer/gstplugins_bad/sys/nvcodec/ |
| D | gstcudaloader.h | 125 CUresult CUDAAPI CuModuleGetFunction (CUfunction* hfunc,
136 CUresult CUDAAPI CuLaunchKernel (CUfunction f,
|
| D | gstcudaloader.c | 100 CUresult (CUDAAPI * CuModuleGetFunction) (CUfunction * hfunc,
106 CUresult (CUDAAPI * CuLaunchKernel) (CUfunction f, unsigned int gridDimX,
473 CuModuleGetFunction (CUfunction * hfunc, CUmodule hmod, const char *name) in CuModuleGetFunction()
500 CuLaunchKernel (CUfunction f, unsigned int gridDimX, in CuLaunchKernel()
|
| D | cuda-converter.c | 742 CUfunction kernel_func[CONVERTER_MAX_NUM_FUNC];
1214 convert_UNPACK_RGB (GstCudaConverter * convert, CUfunction kernel_func, in convert_UNPACK_RGB()
1251 convert_TO_Y444 (GstCudaConverter * convert, CUfunction kernel_func, in convert_TO_Y444()
|
| /third_party/gstreamer/gstplugins_bad/sys/nvcodec/stub/ |
| D | cuda.h | 31 typedef gpointer CUfunction; typedef
|