diff --git a/mace/kernels/eltwise.h b/mace/kernels/eltwise.h index 15f81a3cf76f79261454646b3f23eceec1a6c1a0..b06b267fc71151de03fd36794a481ca6288779a0 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 325f2f414fe96428e376702e93c85edb5f5e9546..bd839c556ede14ffce77e689f0d9476f3134e40e 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);