HomeSort by relevance Sort by last modified time
    Searched refs:smem_usage (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
concat_lib_gpu_impl.cu.cc 152 IntType smem_usage = output_scan.size * sizeof(IntType); local
158 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance)
160 <<<config.block_count, config.thread_per_block, smem_usage,
split_lib_gpu.cu.cc 222 IntType smem_usage = output_scan.size * sizeof(IntType); local
227 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance)
229 <<<config.block_count, config.thread_per_block, smem_usage,

Completed in 68 milliseconds