Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddense_update_functor_gpu.cu.cc56 #define DEFINE_GPU_KERNELS(T) \ macro
66 #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
Dsnapshot_op_gpu.cu.cc28 #define DEFINE_GPU_KERNELS(T) template struct functor::Snapshot<GPUDevice, T>; macro
Dslice_op_gpu.cu.cc30 #define DEFINE_GPU_KERNELS(T) \ macro
Dstrided_slice_op_gpu_impl.h33 #define DEFINE_GPU_KERNELS(T) \ macro
Dbetainc_op_gpu.cu.cc36 #define DEFINE_GPU_KERNELS(T) \ macro
Dsplit_lib_gpu.cu.cc50 #define DEFINE_GPU_KERNELS(T) \ macro
60 #define DEFINE_GPU_KERNELS(T) template struct SplitCustom<Eigen::GpuDevice, T>; macro
Din_topk_op_gpu.cu.cc172 #define DEFINE_GPU_KERNELS(T, TARGET_T) \ macro
Drelu_op_gpu.cu.cc252 #define DEFINE_GPU_KERNELS(T) \ macro
Dmaxpooling_op_gpu.cu.cc470 #define DEFINE_GPU_KERNELS(T) \ macro