Searched refs:kernelParams (Results 1 – 7 of 7) sorted by relevance
/external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
D | ConvertLaunchFuncToRuntimeCalls.cpp | 620 auto kernelParams = generateParamsArray(launchOp, operands, rewriter); in matchAndRewrite() local 627 /*sharedMemBytes=*/zero, stream, kernelParams, in matchAndRewrite()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_9_0.inc | 1166 void **kernelParams, void **extra) { 1173 blockDimZ, sharedMemBytes, hStream, kernelParams, extra); 1180 void **kernelParams) { 1187 blockDimZ, sharedMemBytes, hStream, kernelParams);
|
D | cuda_10_1.inc | 1304 void **kernelParams, void **extra) { 1311 blockDimZ, sharedMemBytes, hStream, kernelParams, extra); 1318 void **kernelParams) { 1325 blockDimZ, sharedMemBytes, hStream, kernelParams);
|
D | cuda_10_0.inc | 1285 void **kernelParams, void **extra) { 1292 blockDimZ, sharedMemBytes, hStream, kernelParams, extra); 1299 void **kernelParams) { 1306 blockDimZ, sharedMemBytes, hStream, kernelParams);
|
D | cuda_10_2.inc | 1425 void **kernelParams, void **extra) { 1432 blockDimZ, sharedMemBytes, hStream, kernelParams, extra); 1439 void **kernelParams) { 1446 blockDimZ, sharedMemBytes, hStream, kernelParams);
|
D | cuda_11_0.inc | 1465 void **kernelParams, void **extra) { 1472 blockDimZ, sharedMemBytes, hStream, kernelParams, extra); 1479 void **kernelParams) { 1486 blockDimZ, sharedMemBytes, hStream, kernelParams);
|
D | cuda_11_2.inc | 1670 void **kernelParams, void **extra) { 1677 blockDimZ, sharedMemBytes, hStream, kernelParams, extra); 1684 void **kernelParams) { 1691 blockDimZ, sharedMemBytes, hStream, kernelParams);
|