Searched refs:kernelParams (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/tests/ut/cpp/stub/runtime/ |
D | cuda.cc | 29 … unsigned int sharedMemBytes, CUstream hStream, void **kernelParams, void **extra) { in cuLaunchKernel() argument
|
D | cuda.h | 78 … unsigned int sharedMemBytes, CUstream hStream, void **kernelParams, void **extra);
|
/third_party/gstreamer/gstplugins_bad/sys/nvcodec/ |
D | gstcudaloader.c | 109 unsigned int sharedMemBytes, CUstream hStream, void **kernelParams, 503 unsigned int sharedMemBytes, CUstream hStream, void **kernelParams, in CuLaunchKernel() argument 509 blockDimX, blockDimY, blockDimZ, sharedMemBytes, hStream, kernelParams, in CuLaunchKernel()
|
D | gstcudaloader.h | 145 void **kernelParams,
|
/third_party/mindspore/mindspore/lite/src/train/ |
D | train_session.cc | 703 auto kernelParams = optimizer->GetOptimizerParams(); in GetOptimizerParams() local 704 for (size_t ix = 0; ix < kernelParams.size(); ix++) { in GetOptimizerParams() 705 auto kernelParam = kernelParams[ix]; in GetOptimizerParams()
|