Home
last modified time | relevance | path

Searched refs:smem_usage (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc153 IntType smem_usage = output_scan.size * sizeof(IntType); in ConcatGPUImpl() local
159 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance) in ConcatGPUImpl()
161 <<<config.block_count, config.thread_per_block, smem_usage, in ConcatGPUImpl()
Dsplit_lib_gpu.cu.cc227 IntType smem_usage = output_scan.size * sizeof(IntType); in Run() local
232 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance) in Run()
234 <<<config.block_count, config.thread_per_block, smem_usage, in Run()