#include "Sme2ShapePolicy.hpp"
#include "Graph.hpp"
#include "Layer.hpp"
#include "armnnUtils/DataLayoutIndexed.hpp"
#include <armnn/Descriptors.hpp>
#include <armnn/Tensor.hpp>
#include <algorithm>
#include <cstdint>

Include dependency graph for Sme2ShapePolicy.cpp:

Namespaces
	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

Functions
void	ApplySme2ShapePolicy (const Graph &graph, bool reduceFp32ToFp16, ModelOptions &modelOptions)

Variable Documentation

◆ m_DepthwiseConvolution2dOps

unsigned int m_DepthwiseConvolution2dOps = 0

Definition at line 26 of file Sme2ShapePolicy.cpp.

◆ m_GemmLikeOps

unsigned int m_GemmLikeOps = 0

Definition at line 25 of file Sme2ShapePolicy.cpp.

◆ m_GemmMacs

int64_t m_GemmMacs = 0

Definition at line 28 of file Sme2ShapePolicy.cpp.

◆ m_HasFp16

bool m_HasFp16 = false

Definition at line 30 of file Sme2ShapePolicy.cpp.

◆ m_HasPoseShape

bool m_HasPoseShape = false

Definition at line 34 of file Sme2ShapePolicy.cpp.

◆ m_HasQuantized

bool m_HasQuantized = false

Definition at line 31 of file Sme2ShapePolicy.cpp.

◆ m_HasSegmentationShape

bool m_HasSegmentationShape = false

Definition at line 32 of file Sme2ShapePolicy.cpp.

◆ m_HasSmallMLargeNProjection

bool m_HasSmallMLargeNProjection = false

Definition at line 35 of file Sme2ShapePolicy.cpp.

◆ m_HasStyleTransferShape

bool m_HasStyleTransferShape = false

Definition at line 33 of file Sme2ShapePolicy.cpp.

◆ m_NonPointwiseGemmMacs

int64_t m_NonPointwiseGemmMacs = 0

Definition at line 29 of file Sme2ShapePolicy.cpp.

◆ m_SmallDenseProjectionOps

unsigned int m_SmallDenseProjectionOps = 0

Definition at line 27 of file Sme2ShapePolicy.cpp.

Namespaces

Functions