Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc158 const int32 kMaxSmemBytesPerformance = 16384; in ConcatGPUImpl() local
159 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance) in ConcatGPUImpl()
Dsplit_lib_gpu.cu.cc231 const int32 kMaxSmemBytesPerformance = 16384; in Run() local
232 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance) in Run()