Searched refs:dynamic_shared_memory_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_launch_config.h | 151 size_t dynamic_shared_memory_size, in GetCudaLaunchConfig() argument
159 &block_count, &thread_per_block, func, dynamic_shared_memory_size, in GetCudaLaunchConfig()
179 size_t dynamic_shared_memory_size, int fixed_block_size) { in GetCudaLaunchConfigFixedBlockSize() argument
185 &block_count, func, fixed_block_size, dynamic_shared_memory_size); in GetCudaLaunchConfigFixedBlockSize()
238 size_t dynamic_shared_memory_size, int block_size_limit) { in GetCuda3DLaunchConfig() argument
259 &block_count, &thread_per_block, func, dynamic_shared_memory_size, in GetCuda3DLaunchConfig()
285 size_t dynamic_shared_memory_size, int block_size_limit) { in GetCuda2DLaunchConfig() argument
287 dynamic_shared_memory_size, block_size_limit); in GetCuda2DLaunchConfig()
|