Home
last modified time | relevance | path

Searched refs:SplitCustom (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsplit_lib_gpu.cu.cc43 void SplitCustom<Device, T>::operator()( in operator ()()
64 #define DEFINE_GPU_KERNELS(T) template struct SplitCustom<Eigen::GpuDevice, T>;
Dsplit_lib.h27 struct SplitCustom { struct
Dsplit_v_op.cc430 functor::SplitCustom<GPUDevice, T>()( in Compute()