From ef882a3bcd2d7449b501fc9d47e88d6f68053a18 Mon Sep 17 00:00:00 2001 From: Unknown Date: Thu, 29 Mar 2018 15:54:42 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E6=A3=80=E6=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- mace/kernels/eltwise.h | 2 +- mace/kernels/opencl/cwise_opencl.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mace/kernels/eltwise.h b/mace/kernels/eltwise.h index c6019f5a..cc3910f7 100644 --- a/mace/kernels/eltwise.h +++ b/mace/kernels/eltwise.h @@ -52,7 +52,7 @@ struct EltwiseFunctor : EltwiseFunctorBase { case PROD: #pragma omp parallel for for (index_t i = 0; i < size; ++i) { - output_ptr[i] = input0_ptr[i] * input1_ptr[i]; + output_ptr[i] = input0_ptr[i] * input1_ptr[i]; } break; case SUM: diff --git a/mace/kernels/opencl/cwise_opencl.cc b/mace/kernels/opencl/cwise_opencl.cc index 325f2f41..bd839c55 100644 --- a/mace/kernels/opencl/cwise_opencl.cc +++ b/mace/kernels/opencl/cwise_opencl.cc @@ -12,8 +12,8 @@ namespace kernels { template void CWiseFunctor::operator()(const Tensor *input, - Tensor *output, - StatsFuture *future) { + Tensor *output, + StatsFuture *future) { const index_t batch = input->dim(0); const index_t height = input->dim(1); const index_t width = input->dim(2); -- GitLab