Searched refs:ConcatCPU (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.cc | 99 void ConcatCPU( in ConcatCPU() function
109 template void ConcatCPU<T>( \
|
| D | concat_lib.h | 45 void ConcatCPU(
|
| D | pack_op.cc | 115 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| D | ragged_tensor_to_variant_op.cc | 238 ConcatCPU<VALUE_TYPE>(context->device(), pieces, &out_flat); in Compute()
|
| D | concat_op.cc | 168 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| D | list_kernels.h | 153 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
464 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
640 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|
| D | tensor_array_ops.cc | 705 ConcatCPU<T>(ctx->device(), input_tensors_flat, &output_flat); in Compute()
936 ConcatCPU<T>(ctx->device(), input_tensors_flat, &output_flat); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | concat_split_util.h | 87 ConcatCPU<T>(context->device(), inputs_flat, &output_flat); in Concat()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 214 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
|