Searched refs:gridZ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_rocm_runtime_wrappers.cc | 96 intptr_t gridZ, intptr_t blockX, in tfKernelGenLaunchKernel() argument
100 if (!gridX || !gridY || !gridZ) { in tfKernelGenLaunchKernel()
113 hipModuleLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, in tfKernelGenLaunchKernel()
|
| D | tf_cuda_runtime_wrappers.cc | 95 intptr_t gridZ, intptr_t blockX, in tfKernelGenLaunchKernel() argument
99 if (!gridX || !gridY || !gridZ) { in tfKernelGenLaunchKernel()
112 cuLaunchKernel(function, gridX, gridY, gridZ, blockX, blockY, blockZ, in tfKernelGenLaunchKernel()
|
| /external/llvm-project/mlir/tools/mlir-cuda-runner/ |
| D | cuda-runtime-wrappers.cpp | 65 intptr_t gridY, intptr_t gridZ, in mgpuLaunchKernel() argument
69 CUDA_REPORT_IF_ERROR(cuLaunchKernel(function, gridX, gridY, gridZ, blockX, in mgpuLaunchKernel()
|
| /external/llvm-project/mlir/tools/mlir-rocm-runner/ |
| D | rocm-runtime-wrappers.cpp | 65 intptr_t gridY, intptr_t gridZ, in mgpuLaunchKernel() argument
70 HIP_REPORT_IF_ERROR(hipModuleLaunchKernel(function, gridX, gridY, gridZ, in mgpuLaunchKernel()
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | cupti_tracer.cc | 568 event.kernel_info.grid_z = kernel->gridZ; in AddKernelActivityEvent()
|