Searched refs:ConcatGPU (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib.h | 52 void ConcatGPU(
60 extern template void ConcatGPU<T>( \
|
| D | concat_lib_gpu.cc | 71 void ConcatGPU( in ConcatGPU() function
95 template void ConcatGPU<T>( \
|
| D | pack_op.cc | 117 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
|
| D | list_kernels.h | 170 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
483 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
659 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
|
| D | concat_op.cc | 159 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
|
| D | tensor_array_ops.cc | 706 ConcatGPU<T>(ctx, input_tensors_flat, output_tensor, &output_flat); in Compute()
938 ConcatGPU<T>(ctx, input_tensors_flat, output_tensor, &output_flat); in Compute()
|
| D | batch_kernels.cc | 87 ConcatGPU<T>(context, inputs_flat, output, &output_flat); in Concat()
|