40 NeonWorkloadFactoryHelper::GetFactory(NeonWorkloadFactoryHelper::GetMemoryManager());
42 unsigned int inputWidth = 2000u;
43 unsigned int inputHeight = 2000u;
44 unsigned int inputChannels = 1u;
45 unsigned int inputBatchSize = 1u;
47 float upperBound = 1.0f;
48 float lowerBound = -1.0f;
50 size_t inputSize = inputWidth * inputHeight * inputChannels * inputBatchSize;
51 std::vector<float> inputData(inputSize, 0.f);
52 std::generate(inputData.begin(), inputData.end(), [](){
53 return (static_cast<float>(rand()) /
static_cast<float>(RAND_MAX / 3)) + 1.f; });
55 unsigned int outputWidth = inputWidth;
56 unsigned int outputHeight = inputHeight;
57 unsigned int outputChannels = inputChannels;
58 unsigned int outputBatchSize = inputBatchSize;
60 armnn::TensorInfo inputTensorInfo({ inputBatchSize, inputChannels, inputHeight, inputWidth },
63 armnn::TensorInfo outputTensorInfo({ outputBatchSize, outputChannels, outputHeight, outputWidth },
68 auto input = MakeTensor<float, 4>(inputTensorInfo, inputData);
70 std::unique_ptr<armnn::ITensorHandle> inputHandle = workloadFactory.
CreateTensorHandle(inputTensorInfo);
71 std::unique_ptr<armnn::ITensorHandle> outputHandle = workloadFactory.
CreateTensorHandle(outputTensorInfo);
76 AddInputToWorkload(descriptor, workloadInfo, inputTensorInfo, inputHandle.get());
77 AddOutputToWorkload(descriptor, workloadInfo, outputTensorInfo, outputHandle.get());
83 std::unique_ptr<armnn::IWorkload> workload = workloadFactory.
CreateActivation(descriptor, workloadInfo);
85 inputHandle->Allocate();
86 outputHandle->Allocate();
98 std::vector<Measurement> measurements = neonTimer.GetMeasurements();
101 if (measurements.size() > 1)
103 BOOST_CHECK_EQUAL(measurements[0].m_Name,
"NeonKernelTimer/0: NEFillBorderKernel");
106 std::ostringstream oss;
107 oss <<
"NeonKernelTimer/" << measurements.size()-1 <<
": NEActivationLayerKernel";
108 BOOST_CHECK_EQUAL(measurements[measurements.size()-1].m_Name, oss.str());
109 BOOST_CHECK(measurements[measurements.size()-1].m_Value > 0.0);
LayerDescriptor m_Parameters
BOOST_CHECK(profilingService.GetCurrentState()==ProfilingState::WaitingForAck)
min(a, max(b, input)) ReLu1 & ReLu6.
std::unique_ptr< IWorkload > CreateActivation(const ActivationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
float m_A
Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH).
Contains information about inputs and outputs to a layer.
float m_B
Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).
ActivationFunction m_Function
The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square).
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const override
void CopyDataToITensorHandle(armnn::ITensorHandle *tensorHandle, const void *memory)