Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc152 IntType smem_max = gpu_device.sharedMemPerBlock(); in ConcatGPUImpl() local
159 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance) in ConcatGPUImpl()
Dsplit_lib_gpu.cu.cc226 IntType smem_max = gpu_device.sharedMemPerBlock(); in Run() local
232 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance) in Run()