Searched refs:kernel_params (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_kernel_to_llvm_pass.cc | 196 auto kernel_params = generateParamsArray(launch_op, operands, rewriter); in matchAndRewrite() local
226 adaptor.blockSizeY(), adaptor.blockSizeZ(), kernel_params}); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 265 GpuStreamHandle stream, void** kernel_params, void** extra);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 428 void** kernel_params = const_cast<void**>(args.argument_addresses().data()); in Launch() local
433 kernel_params, nullptr /* = extra */); in Launch()
|
| D | cuda_driver.cc | 522 unsigned int shared_mem_bytes, CUstream stream, void** kernel_params, in LaunchKernel() argument
532 kernel_params, extra), in LaunchKernel()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 402 unsigned int shared_mem_bytes, GpuStreamHandle stream, void** kernel_params, in LaunchKernel() argument
412 shared_mem_bytes, stream, kernel_params, extra), in LaunchKernel()
|