17 using namespace armcomputetensorutils;
26 const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input);
27 const arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output);
28 arm_compute::TensorInfo aclWeights = BuildArmComputeTensorInfo(weights);
29 aclWeights.set_are_values_constant(weights.
IsConstant());
31 arm_compute::TensorInfo aclBiases;
32 arm_compute::TensorInfo* optionalAclBiases =
nullptr;
37 if (!biases.
value().IsConstant())
40 "Arm NN ClFullyConnectedWorkload does not support non constant bias."};
42 aclBiases = BuildArmComputeTensorInfo(biases.
value());
43 aclBiases.set_are_values_constant(biases.
value().IsConstant());
44 optionalAclBiases = &aclBiases;
47 const arm_compute::FullyConnectedLayerInfo fullyConnectedLayerInfo =
50 return arm_compute::CLFullyConnectedLayer::validate(&aclInput,
54 fullyConnectedLayerInfo);
60 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
61 const arm_compute::CLCompileContext& clCompileContext)
66 arm_compute::ICLTensor& input = PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Inputs[0])->GetTensor();
67 arm_compute::ICLTensor& output = PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Outputs[0])->GetTensor();
68 arm_compute::ICLTensor& weights = PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Inputs[1])->GetTensor();
70 weights.info()->set_are_values_constant(
info.m_InputTensorInfos[1].IsConstant());
72 arm_compute::ICLTensor* bias =
nullptr;
75 bias = &PolymorphicDowncast<IClTensorHandle*>(
m_Data.
m_Inputs[2])->GetTensor();
76 bias->info()->set_are_values_constant(
info.m_InputTensorInfos[2].IsConstant());
84 arm_compute::FullyConnectedLayerInfo fc_info =
90 m_FullyConnectedLayer.configure(clCompileContext,