diff --git a/mace/ops/eltwise.cc b/mace/ops/eltwise.cc index 4166f4d11c9e5b4f4b97585f6e7b64716ecac1ae..8d1cae4bf7a656f55d5164044199146347f0ca1e 100644 --- a/mace/ops/eltwise.cc +++ b/mace/ops/eltwise.cc @@ -433,6 +433,7 @@ inline void TensorEltwise(const EltwiseType type, output[i] = std::pow(input0[i], input1[i]); } } else { +#pragma omp parallel for schedule(runtime) for (index_t i = 0; i < size; ++i) { output[i] = std::pow(input1[i], input0[i]); } @@ -552,6 +553,7 @@ inline void TensorScalarEltwise(const EltwiseType type, output[i] = std::pow(input0[i], input1); } } else { +#pragma omp parallel for schedule(runtime) for (index_t i = 0; i < size; ++i) { output[i] = std::pow(input1, input0[i]); }