Home
last modified time | relevance | path

Searched refs:grid_dim_z (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h205 unsigned int grid_dim_z, unsigned int block_dim_x,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc437 unsigned int grid_dim_y, unsigned int grid_dim_z, unsigned int block_dim_x, in LaunchKernel() argument
443 << " gdy: " << grid_dim_y << " gdz: " << grid_dim_z in LaunchKernel()
447 function, grid_dim_x, grid_dim_y, grid_dim_z, block_dim_x, block_dim_y, in LaunchKernel()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc551 unsigned int grid_dim_y, unsigned int grid_dim_z, unsigned int block_dim_x, in LaunchKernel() argument
557 << " gdy: " << grid_dim_y << " gdz: " << grid_dim_z in LaunchKernel()
560 CUresult res = cuLaunchKernel(function, grid_dim_x, grid_dim_y, grid_dim_z, in LaunchKernel()