23.05
|
Go to the documentation of this file.
12 #include <fmt/format.h>
16 namespace optimizations
27 AddPermuteLayer(graph, PolymorphicDowncast<DepthwiseConvolution2dLayer*>(&layer));
60 PermuteLayer* permuteLayer =
62 PermuteDescriptor(permutationVector),
64 permuteLayer->GetOutputSlot().SetTensorInfo(weightsPermuted);
72 throw InvalidArgumentException(fmt::format(
"Unknown data layout for tensor info conversion: {}",
PermuteDepthwiseConv2dWeightsImpl()=default
void Run(Graph &graph, Layer &layer) const
LayerT * InsertNewLayer(InputSlot &insertBefore, Args &&... args)
Inserts a new layer between the output slot currently connected to insertBefore and insertBefore itse...
Copyright (c) 2021 ARM Limited and Contributors.
const TensorInfo & GetTensorInfo() const override
armnn::TensorShape Permuted(const armnn::TensorShape &srcShape, const armnn::PermutationVector &mappings)
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
LayerType GetType() const override
Returns the armnn::LayerType of this layer.
~PermuteDepthwiseConv2dWeightsImpl()=default
const BackendId & GetBackendId() const
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
This layer represents a depthwise convolution 2d operation.
const Parameters & GetParameters() const override
If the layer has a descriptor return it.
constexpr const char * GetDataLayoutName(DataLayout dataLayout)