Home
last modified time | relevance | path

Searched defs:kernel_params (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc196 auto kernel_params = generateParamsArray(launch_op, operands, rewriter); in matchAndRewrite() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc428 void** kernel_params = const_cast<void**>(args.argument_addresses().data()); in Launch() local
Dcuda_driver.cc522 unsigned int shared_mem_bytes, CUstream stream, void** kernel_params, in LaunchKernel()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc402 unsigned int shared_mem_bytes, GpuStreamHandle stream, void** kernel_params, in LaunchKernel()