提交 c23719f2 编写于 作者: M Michal W. Tarnowski 提交者: Liangliang He

Added missing pragmas (#312)

上级 a1dcdcb5
......@@ -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]);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册