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.cc99 if (shared_mem_size < d.sharedMemPerBlock() && in Compute()
Dconcat_lib_gpu_impl.cu.cc152 IntType smem_max = gpu_device.sharedMemPerBlock(); in ConcatGPUImpl()
Dsplit_lib_gpu.cu.cc226 IntType smem_max = gpu_device.sharedMemPerBlock(); in Run()
Dbias_op_gpu.cu.cc210 const int max_shared_memory_size = d.sharedMemPerBlock() / 2; in compute()
Ddepthwise_conv_op_gpu.h1596 if (shared_memory_size > device.sharedMemPerBlock()) {