Searched refs:ConcatGPUSlice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_gpu.h | 33 void ConcatGPUSlice(
48 extern template void ConcatGPUSlice<T, int32>( \
53 extern template void ConcatGPUSlice<T, int64>( \
|
| D | concat_lib_gpu.cc | 78 ConcatGPUSlice<T, int32>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
80 ConcatGPUSlice<T, int64>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
|
| D | concat_lib_gpu_impl.cu.cc | 116 void ConcatGPUSlice( in ConcatGPUSlice() function
175 template void ConcatGPUSlice<T, int32>( \
182 template void ConcatGPUSlice<T, int64>( \
|