718c41634f
1.项目后端整体迁移至PaddleOCR-NCNN算法，已通过基本的兼容性测试 2.工程改为使用CMake组织，后续为了更好地兼容第三方库，不再提供QMake工程 3.重整权利声明文件，重整代码工程，确保最小化侵权风险 Log: 切换后端至PaddleOCR-NCNN，切换工程为CMake Change-Id: I4d5d2c5d37505a4a24b389b1a4c5d12f17bfa38c
134 lines
3.0 KiB
TableGen
134 lines
3.0 KiB
TableGen
// Tencent is pleased to support the open source community by making ncnn available.
|
|
//
|
|
// Copyright (C) 2020 THL A29 Limited, a Tencent company. All rights reserved.
|
|
//
|
|
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
|
|
// in compliance with the License. You may obtain a copy of the License at
|
|
//
|
|
// https://opensource.org/licenses/BSD-3-Clause
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software distributed
|
|
// under the License is distributed on an "AS IS" BASIS, 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.
|
|
|
|
#ifndef NCNN_OPS_TD
|
|
#define NCNN_OPS_TD
|
|
|
|
include "mlir/IR/OpBase.td"
|
|
include "mlir/Interfaces/SideEffectInterfaces.td"
|
|
include "tf_op_base.td"
|
|
|
|
def NCNN_Dialect : Dialect {
|
|
let name = "ncnn";
|
|
let cppNamespace = "ncnn";
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// NCNN op definitions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
class NCNN_Op<string mnemonic, list<OpTrait> traits = []> :
|
|
Op<NCNN_Dialect, mnemonic, traits>;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// NCNN operations
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
def NCNN_KerasConv2DOp : NCNN_Op<"KerasConv2D", [NoSideEffect]> {
|
|
|
|
let arguments = (ins
|
|
F32Tensor:$x,
|
|
F32Tensor:$weight,
|
|
F32Tensor:$bias,
|
|
|
|
I64ArrayAttr:$strides,
|
|
TF_AnyStrAttrOf<["SAME", "VALID", "EXPLICIT"]>:$padding,
|
|
DefaultValuedAttr<I64ArrayAttr, "{}">:$explicit_paddings,
|
|
DefaultValuedAttr<I64ArrayAttr, "{1, 1, 1, 1}">:$dilations
|
|
);
|
|
|
|
let results = (outs
|
|
F32Tensor:$y
|
|
);
|
|
}
|
|
|
|
def NCNN_KerasDenseOp : NCNN_Op<"KerasDense", [NoSideEffect]> {
|
|
|
|
let arguments = (ins
|
|
F32Tensor:$x,
|
|
F32Tensor:$weight,
|
|
F32Tensor:$bias
|
|
);
|
|
|
|
let results = (outs
|
|
F32Tensor:$y
|
|
);
|
|
}
|
|
|
|
def NCNN_KerasBatchNormOp : NCNN_Op<"KerasBatchNorm", [NoSideEffect]> {
|
|
|
|
let arguments = (ins
|
|
F32Tensor:$x,
|
|
F32Tensor:$gamma,
|
|
F32Tensor:$bias
|
|
);
|
|
|
|
let results = (outs
|
|
F32Tensor:$y
|
|
);
|
|
}
|
|
|
|
def NCNN_BinaryOpOp : NCNN_Op<"BinaryOp", [NoSideEffect]> {
|
|
|
|
let arguments = (ins
|
|
F32Tensor:$x,
|
|
I32Attr:$op_type,
|
|
I32Attr:$with_scalar,
|
|
F32Attr:$b
|
|
);
|
|
|
|
let results = (outs
|
|
F32Tensor:$y
|
|
);
|
|
}
|
|
|
|
def NCNN_InstanceNormOp : NCNN_Op<"InstanceNorm", [NoSideEffect]> {
|
|
|
|
let arguments = (ins
|
|
F32Tensor:$x,
|
|
F32Attr:$epsilon
|
|
);
|
|
|
|
let results = (outs
|
|
F32Tensor:$y
|
|
);
|
|
}
|
|
|
|
def NCNN_InstanceNormAffineOp : NCNN_Op<"InstanceNormAffine", [NoSideEffect]> {
|
|
|
|
let arguments = (ins
|
|
F32Tensor:$x,
|
|
F32Tensor:$gamma,
|
|
F32Tensor:$beta,
|
|
F32Attr:$epsilon
|
|
);
|
|
|
|
let results = (outs
|
|
F32Tensor:$y
|
|
);
|
|
}
|
|
|
|
def NCNN_SwishOp : NCNN_Op<"Swish", [NoSideEffect]> {
|
|
|
|
let arguments = (ins
|
|
F32Tensor:$x
|
|
);
|
|
|
|
let results = (outs
|
|
F32Tensor:$y
|
|
);
|
|
}
|
|
|
|
#endif // NCNN_OPS_TD
|