Searched refs:dynamic_shared_memory_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_launch_config.h | 158 size_t dynamic_shared_memory_size, in GetGpuLaunchConfig() argument
167 &block_count, &thread_per_block, func, dynamic_shared_memory_size, in GetGpuLaunchConfig()
172 &block_count, &thread_per_block, func, dynamic_shared_memory_size, in GetGpuLaunchConfig()
194 size_t dynamic_shared_memory_size, int fixed_block_size) { in GetGpuLaunchConfigFixedBlockSize() argument
201 &block_count, func, fixed_block_size, dynamic_shared_memory_size); in GetGpuLaunchConfigFixedBlockSize()
205 &block_count, func, fixed_block_size, dynamic_shared_memory_size); in GetGpuLaunchConfigFixedBlockSize()
270 size_t dynamic_shared_memory_size, in GetGpu3DLaunchConfig() argument
300 &block_count, &thread_per_block, func, dynamic_shared_memory_size, in GetGpu3DLaunchConfig()
342 size_t dynamic_shared_memory_size, in GetGpu2DLaunchConfig() argument
345 dynamic_shared_memory_size, block_size_limit); in GetGpu2DLaunchConfig()
[all …]
|