ArmNN
 20.05
Pad.cpp File Reference
#include "Pad.hpp"
#include <backendsCommon/WorkloadData.hpp>
#include "TensorBufferArrayView.hpp"
#include "Encoders.hpp"
#include <boost/numeric/conversion/cast.hpp>
#include <cmath>
#include <cstddef>
#include <functional>
#include <limits>
#include <cassert>

Go to the source code of this file.

Namespaces

 armnn
 Copyright (c) 2020 ARM Limited.
 

Functions

template<typename T >
void Pad (const TensorInfo &inputInfo, const TensorInfo &outputInfo, std::vector< std::pair< unsigned int, unsigned int >> m_padList, const T *inputData, T *outData, const float padValue)
 
template void Pad< BFloat16 > (const TensorInfo &inputInfo, const TensorInfo &outputInfo, std::vector< std::pair< unsigned int, unsigned int >> m_PadList, const BFloat16 *inputData, BFloat16 *outData, const float padValue)
 
template void Pad< float > (const TensorInfo &inputInfo, const TensorInfo &outputInfo, std::vector< std::pair< unsigned int, unsigned int >> m_PadList, const float *inputData, float *outData, const float padValue)
 
template void Pad< Half > (const TensorInfo &inputInfo, const TensorInfo &outputInfo, std::vector< std::pair< unsigned int, unsigned int >> m_PadList, const Half *inputData, Half *outData, const float padValue)
 
template void Pad< uint8_t > (const TensorInfo &inputInfo, const TensorInfo &outputInfo, std::vector< std::pair< unsigned int, unsigned int >> m_PadList, const uint8_t *inputData, uint8_t *outData, const float padValue)
 
template void Pad< int8_t > (const TensorInfo &inputInfo, const TensorInfo &outputInfo, std::vector< std::pair< unsigned int, unsigned int >> m_PadList, const int8_t *inputData, int8_t *outData, const float padValue)
 
template void Pad< int16_t > (const TensorInfo &inputInfo, const TensorInfo &outputInfo, std::vector< std::pair< unsigned int, unsigned int >> m_PadList, const int16_t *inputData, int16_t *outData, const float padValue)