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

/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_gpu_impl.cu.cc157 const int32 kMaxSmemBytesPerformance = 16384; local
158 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance)
H A Dsplit_lib_gpu.cu.cc226 const int32 kMaxSmemBytesPerformance = 16384; local
227 if (smem_usage < smem_max && smem_usage < kMaxSmemBytesPerformance)

Completed in 228 milliseconds