提交 d1da472a 编写于 作者: Z zhaojiaying01

Revert: accelerate with openmp

上级 b6e96f27
......@@ -12,20 +12,23 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
#pragma once
#ifdef LRN_OP
#include <cmath>
#include "framework/operator.h"
#include "operators/op_param.h"
#include <cmath>
#ifdef __ARM_NEON
#include <arm_neon.h>
#include "arm_neon.h"
#include "operators/math/math_func_neon.h"
#endif
namespace paddle_mobile {
namespace operators {
using namespace framework;
template <typename T>
struct LRNFunctor {
void operator()(const framework::Tensor &input, framework::Tensor *out, int N,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册