diff --git a/onnxruntime/core/providers/acl/nn/fused_conv.cc b/onnxruntime/core/providers/acl/nn/fused_conv.cc index 740e130915..a10c99ceb8 100644 --- a/onnxruntime/core/providers/acl/nn/fused_conv.cc +++ b/onnxruntime/core/providers/acl/nn/fused_conv.cc @@ -12,18 +12,17 @@ #include "core/providers/acl/acl_common.h" #include "core/providers/acl/acl_fwd.h" #include "core/providers/acl/acl_execution_provider.h" +#include "contrib_ops/cpu/fused_activation.h" namespace onnxruntime { namespace acl{ -template -class FusedConv final : public acl::Conv { - public: - explicit FusedConv(const OpKernelInfo& info) : acl::Conv(info) { +class FusedConv final : public acl::Conv { +public: + explicit FusedConv(const OpKernelInfo& info) : acl::Conv(info) { ORT_ENFORCE(info.GetAttr("activation", &(this->activation_type)).IsOK()); - // printf("fused\n"); + ORT_ENFORCE(GetFusedActivationAttr(info, activation_).IsOK()); } - // Status Compute(OpKernelContext* context) const override; }; ONNX_OPERATOR_TYPED_KERNEL_EX( @@ -33,7 +32,7 @@ ONNX_OPERATOR_TYPED_KERNEL_EX( float, kAclExecutionProvider, KernelDefBuilder().TypeConstraint("T", DataTypeImpl::GetTensorType()), - FusedConv); + FusedConv); } // namespace acl } // namespace onnxruntime