HomeSort by relevance Sort by last modified time
    Searched refs:SplitOpGPULaunch (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
split_lib_gpu.h 33 struct SplitOpGPULaunch {
49 extern template struct SplitOpGPULaunch<T>; \
split_lib_gpu.cu.cc 198 void SplitOpGPULaunch<T>::Run(const Eigen::GpuDevice& d, const T* input,
245 #define REGISTER_GPU_KERNEL(T) template struct SplitOpGPULaunch<T>;
split_op.cc 319 SplitOpGPULaunch<T>().Run(context->eigen_device<GPUDevice>(),

Completed in 89 milliseconds