diff --git a/mace/kernels/eltwise.h b/mace/kernels/eltwise.h index c6019f5acd0515f91d284679e62367756d55b1c5..cc3910f77ceaccc7a80ba6f356ed4724fe90650a 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);