Searched refs:blockSizeZ (Results 1 – 7 of 7) sorted by relevance
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPU.cpp | 214 Value blockSizeZ = numThreadDims > 2 ? dims[numBlockDims + 2] : constOne; in createLaunch() local
220 blockSizeY, blockSizeZ); in createLaunch()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_kernel_to_llvm_pass.cc | 226 adaptor.blockSizeY(), adaptor.blockSizeZ(), kernel_params}); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 247 Value blockSizeX, Value blockSizeY, Value blockSizeZ) { in build() argument
250 {gridSizeX, gridSizeY, gridSizeZ, blockSizeX, blockSizeY, blockSizeZ}); in build()
|
| /external/llvm-project/mlir/include/mlir/Dialect/GPU/ |
| D | GPUOps.td | 299 Index:$blockSizeX, Index:$blockSizeY, Index:$blockSizeZ,
421 `threads` `in` ` ` `(`$blockSizeX`,` $blockSizeY`,` $blockSizeZ`)`
429 Index:$blockSizeX, Index:$blockSizeY, Index:$blockSizeZ)>,
501 "Value":$blockSizeZ)>
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | ConvertLaunchFuncToRuntimeCalls.cpp | 626 launchOp.blockSizeZ(), in matchAndRewrite()
|
| /external/llvm-project/llvm/docs/ |
| D | NVPTXUsage.rst | 718 unsigned blockSizeZ = 1;
730 blockSizeX, blockSizeY, blockSizeZ,
|
| /external/llvm/docs/ |
| D | NVPTXUsage.rst | 726 unsigned blockSizeZ = 1;
738 blockSizeX, blockSizeY, blockSizeZ,
|