You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.7 KiB
71 lines
2.7 KiB
//
|
|
// Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <armnn/TypesUtils.hpp>
|
|
#include <armnn/Tensor.hpp>
|
|
#include <armnn/Types.hpp>
|
|
#include <armnnUtils/TensorUtils.hpp>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
namespace armnn
|
|
{
|
|
class ITensorHandle;
|
|
} // namespace armnn
|
|
|
|
namespace armnnUtils
|
|
{
|
|
armnn::TensorShape GetTensorShape(unsigned int numberOfBatches,
|
|
unsigned int numberOfChannels,
|
|
unsigned int height,
|
|
unsigned int width,
|
|
const armnn::DataLayout dataLayout);
|
|
|
|
armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches,
|
|
unsigned int numberOfChannels,
|
|
unsigned int height,
|
|
unsigned int width,
|
|
const armnn::DataLayout dataLayout,
|
|
const armnn::DataType dataType);
|
|
|
|
armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches,
|
|
unsigned int numberOfChannels,
|
|
unsigned int depth,
|
|
unsigned int height,
|
|
unsigned int width,
|
|
const armnn::DataLayout dataLayout,
|
|
const armnn::DataType dataType);
|
|
|
|
std::pair<float, float> FindMinMax(armnn::ITensorHandle* tensorHandle);
|
|
|
|
armnn::TensorShape ReduceDims(const armnn::TensorShape& tensorInfo, unsigned int dimensions);
|
|
|
|
armnn::TensorInfo ReduceDims(const armnn::TensorInfo& tensorInfo, unsigned int dimensions);
|
|
|
|
armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis);
|
|
|
|
armnn::TensorShape ExpandDimsToRank(const armnn::TensorShape& tensorShape, unsigned int rank);
|
|
|
|
std::vector<unsigned int> SqueezeDims(const armnn::TensorShape& tensorShape);
|
|
|
|
unsigned int GetNumElementsBetween(const armnn::TensorShape& shape,
|
|
unsigned int firstAxisInclusive,
|
|
unsigned int lastAxisExclusive);
|
|
|
|
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis);
|
|
|
|
unsigned int GetNumElementsAfter(const armnn::TensorShape& shape, unsigned int axis);
|
|
|
|
std::pair<unsigned int, std::vector<float>> GetPerAxisParams(const armnn::TensorInfo& info);
|
|
|
|
template<typename PrimitiveType>
|
|
std::unique_ptr<float[]> ToFloatArray(const std::vector<PrimitiveType>& data, const armnn::TensorInfo& tensorInfo);
|
|
|
|
std::unique_ptr<float[]> ToFloatArray(const std::vector<uint8_t>& data, const armnn::TensorInfo& tensorInfo);
|
|
|
|
} // namespace armnnUtils
|