Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper.h59 #define GPU_DYNAMIC_SHARED_MEM_DECL(ALIGN, TYPE, NAME) \ macro
64 #define GPU_DYNAMIC_SHARED_MEM_DECL(ALIGN, TYPE, NAME) \ macro
/external/tensorflow/tensorflow/core/kernels/
Dbucketize_op_gpu.cu.cc41 GPU_DYNAMIC_SHARED_MEM_DECL(sizeof(float), unsigned char, shared_mem); in BucketizeCustomKernel()
Dsplit_lib_gpu.cu.cc128 GPU_DYNAMIC_SHARED_MEM_DECL(sizeof(T), unsigned char, smem); in split_v_kernel()
Dconcat_lib_gpu_impl.cu.cc73 GPU_DYNAMIC_SHARED_MEM_DECL(sizeof(T), unsigned char, smem); in concat_variable_kernel()
Dbias_op_gpu.cu.cc131 GPU_DYNAMIC_SHARED_MEM_DECL(8, char, s_buf); in BiasGradNHWC_SharedAtomics()
Ddepthwise_conv_op_gpu.h197 GPU_DYNAMIC_SHARED_MEM_DECL(8, unsigned char, shared_memory);
488 GPU_DYNAMIC_SHARED_MEM_DECL(8, unsigned char, shared_memory);
1166 GPU_DYNAMIC_SHARED_MEM_DECL(8, unsigned char, shared_memory);
1439 GPU_DYNAMIC_SHARED_MEM_DECL(8, unsigned char, shared_memory);
Dreduction_gpu_kernels.cu.h559 GPU_DYNAMIC_SHARED_MEM_DECL(8, char, ss);