27 using namespace armnn;
34 Connect(input, resize, inputInfo, 0, 0);
35 Connect(resize, output, outputInfo, 0, 0);
40 template<armnn::DataType ArmnnType>
41 void ResizeEndToEnd(
const std::vector<armnn::BackendId>& backends,
45 using namespace armnn;
48 constexpr
unsigned int inputWidth = 3u;
49 constexpr
unsigned int inputHeight = inputWidth;
51 constexpr
unsigned int outputWidth = 5u;
52 constexpr
unsigned int outputHeight = outputWidth;
57 const float qScale = IsQuantizedType<T>() ? 0.25f : 1.0f;
58 const int32_t qOffset = IsQuantizedType<T>() ? 50 : 0;
60 TensorInfo inputInfo(inputShape, ArmnnType, qScale, qOffset,
true);
61 TensorInfo outputInfo(outputShape, ArmnnType, qScale, qOffset);
63 std::vector<float> inputData =
70 std::vector<float> expectedOutputData;
77 1.0f, 1.6f, 2.2f, 2.8f, 3.0f,
78 2.8f, 3.4f, 4.0f, 4.6f, 4.8f,
79 4.6f, 5.2f, 5.8f, 6.4f, 6.6f,
80 6.4f, 7.0f, 7.6f, 8.2f, 8.4f,
81 7.0f, 7.6f, 8.2f, 8.8f, 9.0f
89 1.f, 1.f, 2.f, 2.f, 3.f,
90 1.f, 1.f, 2.f, 2.f, 3.f,
91 4.f, 4.f, 5.f, 5.f, 6.f,
92 4.f, 4.f, 5.f, 5.f, 6.f,
93 7.f, 7.f, 8.f, 8.f, 9.f
112 constexpr
size_t dataTypeSize =
sizeof(float);
115 std::vector<float> tmp(inputData.size());
121 std::vector<T> qInputData = armnnUtils::QuantizedVector<T>(inputData, qScale, qOffset);
122 std::vector<T> qExpectedOutputData = armnnUtils::QuantizedVector<T>(expectedOutputData, qScale, qOffset);
124 INetworkPtr network = CreateResizeNetwork(descriptor, inputInfo, outputInfo);
126 EndToEndLayerTestImpl<ArmnnType, ArmnnType>(std::move(network),
127 { { 0, qInputData } },
128 { { 0, qExpectedOutputData } },
134 template<armnn::DataType ArmnnType>
141 template<armnn::DataType ArmnnType>
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
const TensorShape & GetShape() const
void ResizeBilinearEndToEnd(const std::vector< armnn::BackendId > &backends, armnn::DataLayout dataLayout)
ResizeMethod m_Method
The Interpolation method to use (Bilinear, NearestNeighbor).
void ResizeNearestNeighborEndToEnd(const std::vector< armnn::BackendId > &backends, armnn::DataLayout dataLayout)
typename ResolveTypeImpl< DT >::Type ResolveType
Copyright (c) 2021 ARM Limited and Contributors.
A ResizeBilinearDescriptor for the ResizeBilinearLayer.
uint32_t m_TargetWidth
Target width value.
uint32_t m_TargetHeight
Target height value.
void Connect(armnn::IConnectableLayer *from, armnn::IConnectableLayer *to, const armnn::TensorInfo &tensorInfo, unsigned int fromIndex, unsigned int toIndex)
armnn::TensorShape MakeTensorShape(unsigned int batches, unsigned int channels, unsigned int height, unsigned int width, armnn::DataLayout layout)
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
static INetworkPtr Create(NetworkOptions networkOptions={})