Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc446 void **kernel_params = const_cast<void **>(args.argument_addresses().data()); in Launch() local
Dcuda_driver.cc553 unsigned int shared_mem_bytes, CUstream stream, void** kernel_params, in LaunchKernel()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc439 unsigned int shared_mem_bytes, GpuStreamHandle stream, void** kernel_params, in LaunchKernel()