14 namespace optimizations
17 template <
typename LayerT>
20 LayerT* layer = PolymorphicDowncast<LayerT*>(l);
56 ConvertWeight<Convolution2dLayer>(&layer);
64 ConvertWeight<FullyConnectedLayer>(&layer);
~ConvertFp32NetworkToBf16Impl()=default
const TensorShape & GetShape() const
Copyright (c) 2020 ARM Limited.
LayerT * ConvertWeight(Layer *l)
DataType GetDataType() const
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
static void ConvertFloat32ToBFloat16(const float *srcFloat32Buffer, size_t numElements, void *dstBFloat16Buffer)
std::vector< ConvertFp32ToBf16Layer * > InsertConvertFp32ToBf16LayersBefore(Graph &graph, Layer &layer, bool expectCorrectInputType)
ConvertFp32NetworkToBf16Impl()=default
DataType GetDataType() const
LayerType GetType() const
void Run(Graph &graph, Layer &layer) const
unsigned int GetNumElements() const