Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc201 auto kernel_params = generateParamsArray(launch_op, operands, rewriter); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dral_legalize_to_llvm.cc498 auto kernel_params = generateParamsArray(launch_op, operands, rewriter); in matchAndRewrite() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc424 void** kernel_params = const_cast<void**>(args.argument_addresses().data()); in Launch() local
Dcuda_driver.cc470 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()