diff options
Diffstat (limited to 'tensorflow/contrib/lite/kernels/internal')
-rw-r--r-- | tensorflow/contrib/lite/kernels/internal/optimized/multithreaded_conv.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/internal/optimized/multithreaded_conv.h b/tensorflow/contrib/lite/kernels/internal/optimized/multithreaded_conv.h index 5fb31889fe..59f0e3c927 100644 --- a/tensorflow/contrib/lite/kernels/internal/optimized/multithreaded_conv.h +++ b/tensorflow/contrib/lite/kernels/internal/optimized/multithreaded_conv.h @@ -113,8 +113,8 @@ class EigenTensorConvFunctor { filter_width * filter_height * input_depth; Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); - EigenMatrix output(output_data, 1, filter_count); - ConstEigenMatrix input(input_data, 1, k); + EigenMatrix output(output_data, input_batches, filter_count); + ConstEigenMatrix input(input_data, input_batches, k); ConstEigenMatrix filter(filter_data, k, filter_count); MatMulConvFunctor<Eigen::ThreadPoolDevice, T>()(device, output, input, filter, dim_pair); |