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.cc202 auto kernel_params = generateParamsArray(launch_op, adaptor, rewriter); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_gpu_executor.cc469 void** kernel_params = const_cast<void**>(args.argument_addresses().data()); in Launch() local
Dcuda_driver.cc464 void** kernel_params, void** extra) { in LaunchKernel()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc409 GpuStreamHandle stream, void** kernel_params, void** extra) { in LaunchKernel()