21 m_Weight = std::make_unique<ScopedTensorHandle>(*(descriptor.
m_Weight));
22 const TensorInfo& rWeightInfo = m_Weight->GetTensorInfo();
23 m_WeightShape = rWeightInfo.
GetShape();
24 m_WeightDecoder = MakeDecoder<float>(rWeightInfo, m_Weight->Map(
true));
28 m_Bias = std::make_unique<ScopedTensorHandle>(*(descriptor.
m_Bias));
29 const TensorInfo& biasInfo = m_Bias->GetTensorInfo();
30 m_BiasDecoder = MakeDecoder<float>(biasInfo, m_Bias->Map(
true));
41 std::vector<ITensorHandle*> outputs)
51 m_WeightShape = rWeightInfo.
GetShape();
52 m_WeightDecoder = MakeDecoder<float>(rWeightInfo);
57 m_BiasDecoder = MakeDecoder<float>(biasInfo);
62 m_OutputShape = outputInfo.GetShape();
67 m_NumActivations *= inputInfo.
GetShape()[i];
87 std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(
GetTensorInfo(inputs[0]), inputs[0]->Map());
88 std::unique_ptr<Encoder<float>> OutputEncoder = MakeEncoder<float>(
GetTensorInfo(outputs[0]), outputs[0]->Map());
92 m_WeightDecoder->Reset(inputs[1]->
Map());
95 m_BiasDecoder->Reset(inputs[2]->
Map());
const TensorShape & GetShape() const
CPU Execution: Reference C++ kernels.
bool m_TransposeWeightMatrix
Enable/disable transpose weight matrix.
Copyright (c) 2021 ARM Limited and Contributors.
const ConstTensorHandle * m_Bias
LayerDescriptor m_Parameters
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
void PostAllocationConfigure() override
std::vector< ITensorHandle * > m_Inputs
void Execute() const override
FullyConnectedQueueDescriptor m_Data
bool m_BiasEnabled
Enable/disable bias.
#define ARMNN_ASSERT(COND)
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
const ConstTensorHandle * m_Weight
RefFullyConnectedWorkload(const FullyConnectedQueueDescriptor &descriptor, const WorkloadInfo &info)
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Outputs
Contains information about inputs and outputs to a layer.
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
unsigned int GetNumDimensions() const
bool m_ConstantWeights
Enable/disable constant weights and biases.