Home
last modified time | relevance | path

Searched defs:reshapeLayer (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DAddBroadcastReshapeLayerTests.cpp57 Layer* const reshapeLayer = GetFirstLayerWithName(graph, reshapeLayerName); in AddBroadcastReshapeLayerOptimizerTest() local
151 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:sub-0"); variable
196 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:div-0"); variable
241 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0"); variable
285 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0"); variable
334 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0"); variable
H A DTransposeAsReshapeTests.cpp48 const auto reshapeLayer = static_cast<const armnn::ReshapeLayer*>(layer); in __anona4a2a7ab0102() local
H A DPermuteAsReshapeTests.cpp48 const auto reshapeLayer = static_cast<const armnn::ReshapeLayer*>(layer); in __anon71ec646e0102() local
H A DOptimizeConsecutiveReshapesTests.cpp50 const auto reshapeLayer = static_cast<const armnn::ReshapeLayer*>(layer); in __anon69a3f7840102() local
/aosp_15_r20/external/armnn/src/armnn/optimizations/
H A DAddBroadcastReshapeLayer.hpp82 ReshapeLayer* reshapeLayer = in Run() local
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DUnpack.hpp193 …armnn::IConnectableLayer* reshapeLayer = delegateData.m_Network->AddReshapeLayer(reshapeDescriptor, in VisitUnpackOperator() local
H A DFullyConnected.hpp216 armnn::IConnectableLayer* reshapeLayer = nullptr; in VisitFullyConnectedOperator() local
H A DClassicDelegateUtils.hpp298 armnn::IConnectableLayer* reshapeLayer = data.m_Network->AddReshapeLayer(desc); in AddReshapeLayer() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DConversionUtils.hpp235 armnn::IConnectableLayer* reshapeLayer = network.AddReshapeLayer(reshapeDescriptor); in AddReshapeLayer() local
341 …armnn::IConnectableLayer& reshapeLayer = AddReshapeLayer(*data.m_Network, smallInputHandle, reshap… in BroadcastTensor() local
H A DConverter.cpp2030 armnn::IConnectableLayer* reshapeLayer = data.m_Network->AddReshapeLayer(reshapeDescriptor); in ConvertFullyConnected() local
/aosp_15_r20/external/armnn/delegate/opaque/src/
H A DOpaqueDelegateUtils.hpp350 armnn::IConnectableLayer* reshapeLayer = data.m_Network->AddReshapeLayer(desc); in AddReshapeLayer() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils.hpp288 armnn::IConnectableLayer* reshapeLayer = network.AddReshapeLayer(reshapeDescriptor); in AddReshapeLayer() local
371 …armnn::IConnectableLayer& reshapeLayer = AddReshapeLayer(*data.m_Network, smallInputHandle, reshap… in BroadcastTensor() local
3219 armnn::IConnectableLayer* reshapeLayer = data.m_Network->AddReshapeLayer(reshapeDescriptor); in ConvertFullyConnected() local
/aosp_15_r20/external/armnn/src/armnnOnnxParser/
H A DOnnxParser.cpp1239 …IConnectableLayer* reshapeLayer = m_Network->AddReshapeLayer(reshapeDescriptor, reshapeLayerName.c… in AddFullyConnected() local
1298 … IConnectableLayer* reshapeLayer = m_Network->AddReshapeLayer(desc, reshapeLayerName.c_str()); in AddFullyConnected() local
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp3362 armnn::IConnectableLayer* reshapeLayer = m_Network->AddReshapeLayer(reshapeDescriptor, in ParseFullyConnected() local
3902 … armnn::IConnectableLayer* reshapeLayer = m_Network->AddReshapeLayer(desc, layerName.c_str()); in ParseUnpack() local
4652 IConnectableLayer* reshapeLayer = in AddReshapeLayer() local
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DSerializerTests.cpp2277 …armnn::IConnectableLayer* const reshapeLayer = network->AddReshapeLayer(descriptor, layerName.c_st… variable