Home
last modified time | relevance | path

Searched defs:DEFINE_GPU_KERNELS (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddense_update_functor_gpu.cu.cc56 #define DEFINE_GPU_KERNELS(T) \ macro
65 #define DEFINE_GPU_KERNELS(T) \ macro
Dsnapshot_op_gpu.cu.cc28 #define DEFINE_GPU_KERNELS(T) template struct functor::Snapshot<GPUDevice, T>; macro
Dstrided_slice_op_gpu_impl.h33 #define DEFINE_GPU_KERNELS(T) \ macro
Dslice_op_gpu.cu.cc30 #define DEFINE_GPU_KERNELS(T) \ macro
Dsoftplus_op_gpu.cu.cc32 #define DEFINE_GPU_KERNELS(T) \ macro
Dsoftsign_op_gpu.cu.cc32 #define DEFINE_GPU_KERNELS(T) \ macro
Dbetainc_op_gpu.cu.cc35 #define DEFINE_GPU_KERNELS(T) \ macro
Davgpooling_op_gpu.cu.cc33 #define DEFINE_GPU_KERNELS(T) \ macro
37 DEFINE_GPU_KERNELS(float) in DEFINE_GPU_KERNELS() function
Dsplit_lib_gpu.cu.cc51 #define DEFINE_GPU_KERNELS(T) \ macro
64 #define DEFINE_GPU_KERNELS(T) template struct SplitCustom<Eigen::GpuDevice, T>; macro
Drelu_op_gpu.cu.cc148 #define DEFINE_GPU_KERNELS(T) \ macro
Dmaxpooling_op_gpu.cu.cc517 #define DEFINE_GPU_KERNELS(T) \ macro