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

  /external/tensorflow/tensorflow/core/kernels/
concat_lib_gpu.cc 34 void ConcatGPUSlice(
92 ConcatGPUSlice<T, int32>(c->eigen_gpu_device(), inputs_flat, output_flat);
94 ConcatGPUSlice<T, int64>(c->eigen_gpu_device(), inputs_flat, output_flat);
concat_lib_gpu_impl.cu.cc 115 void ConcatGPUSlice(
174 template void ConcatGPUSlice<T, int32>( \
181 template void ConcatGPUSlice<T, int64>( \

Completed in 68 milliseconds