diff options
-rw-r--r-- | tensorflow/core/kernels/concat_lib_cpu.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/concat_lib_cpu.cc b/tensorflow/core/kernels/concat_lib_cpu.cc index d02d083372..2ba40f5346 100644 --- a/tensorflow/core/kernels/concat_lib_cpu.cc +++ b/tensorflow/core/kernels/concat_lib_cpu.cc @@ -123,7 +123,7 @@ void ConcatCPU(DeviceBase* d, } } }; - Shard(num_threads, worker_threads->workers, output->size(), 100, work); + Shard(num_threads, worker_threads->workers, output->size(), sizeof(T), work); } #define REGISTER(T) \ |