#include <BFloat16.hpp>
#include <Half.hpp>
#include <aclCommon/ArmComputeTensorUtils.hpp>
#include <armnn/backends/TensorHandle.hpp>
#include <armnn/Utils.hpp>
#include <arm_compute/runtime/CL/CLTensor.h>
#include <arm_compute/runtime/IFunction.h>
#include <sstream>
Go to the source code of this file.
|
| armnn |
| Copyright (c) 2021 ARM Limited and Contributors.
|
|
|
std::string | GetConvolutionMethodString (arm_compute::ConvolutionMethod &convolutionMethod) |
|
template<typename T > |
void | CopyArmComputeClTensorData (arm_compute::CLTensor &dstTensor, const T *srcData) |
|
auto | SetClStridedSliceData (const std::vector< int > &m_begin, const std::vector< int > &m_end, const std::vector< int > &m_stride) |
|
auto | SetClSliceData (const std::vector< unsigned int > &m_begin, const std::vector< unsigned int > &m_size) |
|
void | InitializeArmComputeClTensorData (arm_compute::CLTensor &clTensor, const ConstTensorHandle *handle) |
|
RuntimeException | WrapClError (const cl::Error &clError, const CheckLocation &location) |
|
void | RunClFunction (arm_compute::IFunction &function, const CheckLocation &location) |
|
template<typename DataType , typename PayloadType > |
DataType * | GetOutputTensorData (unsigned int idx, const PayloadType &data) |
|