Home
last modified time | relevance | path

Searched refs:sharedMemPerBlock (Results 1 – 7 of 7) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h287 EIGEN_STRONG_INLINE int sharedMemPerBlock() const { in initializeDeviceProp() function
288 return stream_->deviceProperties().sharedMemPerBlock; in initializeDeviceProp()
DTensorConvolution.h854 const int maxSharedMem = m_device.sharedMemPerBlock();
/external/tensorflow/tensorflow/core/kernels/
Dbucketize_op_gpu.cu.cc98 if (shared_mem_size < d.sharedMemPerBlock() && in Compute()
Dsplit_lib_gpu.cu.cc228 IntType smem_max = gpu_device.sharedMemPerBlock(); in Run()
Dconcat_lib_gpu_impl.cu.cc153 IntType smem_max = gpu_device.sharedMemPerBlock(); in ConcatGPUImpl()
Dbias_op_gpu.cu.cc225 const int max_shared_memory_size = d.sharedMemPerBlock() / 2; in compute()
Ddepthwise_conv_op_gpu.h1592 if (shared_memory_size > device.sharedMemPerBlock()) {