19 #include <arm_compute/runtime/CL/functions/CLDepthwiseConvolutionLayer.h>
24 using namespace armcomputetensorutils;
33 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.
m_DataLayout);
34 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.
m_DataLayout);
45 unsigned int aclDepthMultiplier;
50 arm_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weightsPermuted, descriptor.
m_DataLayout);
51 aclWeightsInfo.set_are_values_constant(weights.
IsConstant());
53 arm_compute::TensorInfo aclBiasesInfo;
54 arm_compute::TensorInfo* optionalAclBiasesInfo =
nullptr;
59 if (!biases.
value().IsConstant())
62 "ArmNN ClDepthwiseConv2dWorkload does not support non constant bias."};
64 aclBiasesInfo = BuildArmComputeTensorInfo(biases.
value(), descriptor.
m_DataLayout);
65 aclBiasesInfo.set_are_values_constant(biases.
value().IsConstant());
66 optionalAclBiasesInfo = &aclBiasesInfo;
69 const arm_compute::PadStrideInfo aclPadStrideInfo = BuildArmComputePadStrideInfo(descriptor);
70 const arm_compute::Size2D aclDilationInfo = BuildArmComputeSize2D(
75 activationDescriptor);
77 return arm_compute::CLDepthwiseConvolutionLayer::validate(&aclInputInfo,
79 optionalAclBiasesInfo,
91 const arm_compute::CLCompileContext& clCompileContext)
96 arm_compute::ICLTensor& input = PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Inputs[0])->GetTensor();
97 arm_compute::ICLTensor& output = PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Outputs[0])->GetTensor();
98 arm_compute::ICLTensor& weights = PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Inputs[1])->GetTensor();
99 weights.info()->set_are_values_constant(
info.m_InputTensorInfos[1].IsConstant());
100 arm_compute::ITensorInfo* weightsInfo = weights.info();
101 arm_compute::ITensorInfo* inputInfo = input.info();
102 auto weightsShape = weightsInfo->tensor_shape();
103 auto inputShape = inputInfo->tensor_shape();
107 unsigned int depthMultiplier =
110 arm_compute::ICLTensor* bias =
nullptr;
113 bias = &PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Inputs[2])->GetTensor();
114 bias->info()->set_are_values_constant(
info.m_InputTensorInfos[2].IsConstant());
119 const arm_compute::Size2D aclDilationInfo = BuildArmComputeSize2D(
124 input.info()->set_data_layout(aclDataLayout);
125 weights.info()->set_data_layout(aclDataLayout);
126 output.info()->set_data_layout(aclDataLayout);
128 arm_compute::PadStrideInfo padStrideInfo = BuildArmComputePadStrideInfo(
m_Data.
m_Parameters);