21 namespace optimizations
26 static void Func(std::unique_ptr<ScopedCpuTensorHandle>& handle)
47 static void Func(std::unique_ptr<ScopedCpuTensorHandle>& handle)
68 static void Func(std::unique_ptr<ScopedCpuTensorHandle>& handle)
89 static void Func(std::unique_ptr<ScopedCpuTensorHandle>& handle)
108 template<
typename Converter,
typename Predicate>
119 if (Predicate::Test(layer))
static bool Test(const Layer &layer)
const TensorShape & GetShape() const
static bool Test(const Layer &layer)
static void ConvertBFloat16ToFloat32(const void *srcBFloat16Buffer, size_t numElements, float *dstFloat32Buffer)
void OperateOnConstantTensors(Op op)
static bool Test(const Layer &layer)
static void Func(std::unique_ptr< ScopedCpuTensorHandle > &handle)
static void Func(std::unique_ptr< ScopedCpuTensorHandle > &handle)
Copyright (c) 2020 ARM Limited.
void IgnoreUnused(Ts &&...)
static void Func(std::unique_ptr< ScopedCpuTensorHandle > &handle)
static void Func(std::unique_ptr< ScopedCpuTensorHandle > &handle)
DataType GetDataType() const
static void ConvertFloat32To16(const float *srcFloat32Buffer, size_t numElements, void *dstFloat16Buffer)
Converts a buffer of FP32 values to FP16, and stores in the given dstFloat16Buffer.
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
static void ConvertFloat16To32(const void *srcFloat16Buffer, size_t numElements, float *dstFloat32Buffer)
void Run(Graph &graph, Layer &layer) const override
static void ConvertFloat32ToBFloat16(const float *srcFloat32Buffer, size_t numElements, void *dstBFloat16Buffer)
DataType GetDataType() const
unsigned int GetNumElements() const