17 #include <arm_compute/runtime/CL/functions/CLConvolutionLayer.h>
21 using namespace armcomputetensorutils;
28 bool isFastMathEnabled,
31 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.
m_DataLayout);
32 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.
m_DataLayout);
33 arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.
m_DataLayout);
34 aclWeightsInfo.set_are_values_constant(weights.
IsConstant());
36 const arm_compute::Size2D aclDilationInfo = BuildArmComputeSize2D(descriptor.
m_DilationX,
39 arm_compute::TensorInfo aclBiasesInfo;
40 arm_compute::TensorInfo *optionalAclBiasesInfo =
nullptr;
46 if (!biases.
value().IsConstant())
49 "ArmNN ClConvolution2dWorkload does not support non constant bias."};
51 aclBiasesInfo = BuildArmComputeTensorInfo(biases.
value(), descriptor.
m_DataLayout);
52 aclBiasesInfo.set_are_values_constant(biases.
value().IsConstant());
53 optionalAclBiasesInfo = &aclBiasesInfo;
56 arm_compute::PadStrideInfo layerInfo = BuildArmComputePadStrideInfo(descriptor);
59 activationDescriptor);
61 return arm_compute::CLConvolutionLayer::validate(&aclInputInfo,
63 optionalAclBiasesInfo,
66 arm_compute::WeightsInfo(),
74 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
75 const arm_compute::CLCompileContext& clCompileContext,
76 const bool isFastMathEnabled)
78 , m_ConvolutionLayer(memoryManager)
91 weights.info()->set_are_values_constant(
info.m_InputTensorInfos[1].IsConstant());
96 bias.info()->set_are_values_constant(
info.m_InputTensorInfos[2].IsConstant());
100 m_BiasProxy = std::make_unique<ICLTensorProxy>(&bias);
104 m_InputProxy = std::make_unique<ICLTensorProxy>(&input);
105 m_OutputProxy = std::make_unique<ICLTensorProxy>(&output);
106 m_WeightsProxy = std::make_unique<ICLTensorProxy>(&weights);
109 input.info()->set_data_layout(aclDataLayout);
110 output.info()->set_data_layout(aclDataLayout);
111 weights.info()->set_data_layout(aclDataLayout);
113 arm_compute::PadStrideInfo padStrideInfo = BuildArmComputePadStrideInfo(
m_Data.
m_Parameters);
119 m_ConvolutionLayer.configure(clCompileContext,
121 m_WeightsProxy.get(),
125 arm_compute::WeightsInfo(),
131 m_ConvolutionMethod =
132 m_ConvolutionLayer.get_convolution_method(input.info(),
136 arm_compute::WeightsInfo(),
138 arm_compute::CLScheduler::get().target(),
169 return m_ConvolutionMethod;
177 m_InputProxy->set(&input);
178 m_OutputProxy->set(&output);