Home
last modified time | relevance | path

Searched refs:GPU_DYNAMIC_SHARED_MEM_DECL (Results 1 – 8 of 8) 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()
Dbincount_op_gpu.cu.cc160 GPU_DYNAMIC_SHARED_MEM_DECL(sizeof(T), unsigned char, shared_col_mem); in BincountColReduceSharedKernel()
Dconcat_lib_gpu_impl.cu.cc73 GPU_DYNAMIC_SHARED_MEM_DECL(sizeof(T), unsigned char, smem); in concat_variable_kernel()
Dsplit_lib_gpu.cu.cc124 GPU_DYNAMIC_SHARED_MEM_DECL(sizeof(T), unsigned char, smem); in split_v_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);
1168 GPU_DYNAMIC_SHARED_MEM_DECL(8, unsigned char, shared_memory);
1441 GPU_DYNAMIC_SHARED_MEM_DECL(8, unsigned char, shared_memory);
Dreduction_gpu_kernels.cu.h555 GPU_DYNAMIC_SHARED_MEM_DECL(8, char, ss);