Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu.h40 void ConcatGPUImpl(const Eigen::GpuDevice& d,
58 extern template void ConcatGPUImpl<T, int32>( \
63 extern template void ConcatGPUImpl<T, int64>( \
Dconcat_lib_gpu_impl.cu.cc138 void ConcatGPUImpl(const Eigen::GpuDevice& gpu_device, in ConcatGPUImpl() function
189 template void ConcatGPUImpl<T, int32>( \
196 template void ConcatGPUImpl<T, int64>( \
Dconcat_lib_gpu.cc63 ConcatGPUImpl<T, IntType>(c->eigen_gpu_device(), input_ptrs.data(), in ConcatGPUCall()