Home
last modified time | relevance | path

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

/external/armnn/src/armnn/test/optimizations/
DAddBroadcastReshapeLayerTests.cpp56 Layer* const reshapeLayer = GetFirstLayerWithName(graph, reshapeLayerName); in AddBroadcastReshapeLayerOptimizerTest() local
150 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:sub-0"); in BOOST_AUTO_TEST_CASE() local
195 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:div-0"); in BOOST_AUTO_TEST_CASE() local
240 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0"); in BOOST_AUTO_TEST_CASE() local
284 Layer* const reshapeLayer = GetFirstLayerWithName(graph, "Reshape_for:mul-0"); in BOOST_AUTO_TEST_CASE() local
DTransposeAsReshapeTests.cpp47 const auto reshapeLayer = static_cast<const armnn::ReshapeLayer*>(layer); in BOOST_AUTO_TEST_CASE() local
DPermuteAsReshapeTests.cpp47 const auto reshapeLayer = static_cast<const armnn::ReshapeLayer*>(layer); in BOOST_AUTO_TEST_CASE() local
DOptimizeConsecutiveReshapesTests.cpp49 const auto reshapeLayer = static_cast<const armnn::ReshapeLayer*>(layer); in BOOST_AUTO_TEST_CASE() local
/external/armnn/src/armnn/optimizations/
DAddBroadcastReshapeLayer.hpp70 … ReshapeLayer *reshapeLayer = graph.InsertNewLayer<ReshapeLayer>(layer.GetInputSlot(reshapeSlot), in Run() local
/external/armnn/delegate/src/
DComparison.hpp117 auto reshapeLayer = BroadcastTensor(inputTensorInfo0, in VisitComparisonOperator() local
DFullyConnected.hpp204 armnn::IConnectableLayer* reshapeLayer = nullptr; in VisitFullyConnectedOperator() local
DElementwiseBinary.hpp304 auto reshapeLayer = BroadcastTensor(inputTensorInfo0, in VisitElementwiseBinaryOperator() local
DDelegateUtils.hpp216 …armnn::IConnectableLayer* reshapeLayer = delegateData.m_Network->AddReshapeLayer(reshapeDescriptor… in BroadcastTensor() local
/external/android-nn-driver/
DConversionUtils.hpp281 armnn::IConnectableLayer* reshapeLayer = network.AddReshapeLayer(reshapeDescriptor); in AddReshapeLayer() local
352 …armnn::IConnectableLayer& reshapeLayer = AddReshapeLayer(*data.m_Network, smallInputHandle, reshap… in BroadcastTensor() local
3123 armnn::IConnectableLayer* reshapeLayer = data.m_Network->AddReshapeLayer(reshapeDescriptor); in ConvertFullyConnected() local
/external/armnn/src/armnnTfLiteParser/
DTfLiteParser.cpp2346 … armnn::IConnectableLayer* reshapeLayer = m_Network->AddReshapeLayer(desc, layerName.c_str()); in ParseFullyConnected() local
2576 … armnn::IConnectableLayer* reshapeLayer = m_Network->AddReshapeLayer(desc, layerName.c_str()); in ParseUnpack() local
/external/armnn/src/armnnTfParser/
DTfParser.cpp274 …IConnectableLayer* const reshapeLayer = m_Network.AddReshapeLayer(reshapeDesc, reshapeLayerName.c_… in AddBroadcastReshapeLayer() local
/external/armnn/src/armnnSerializer/test/
DSerializerTests.cpp2311 …armnn::IConnectableLayer* const reshapeLayer = network->AddReshapeLayer(descriptor, layerName.c_st… in BOOST_AUTO_TEST_CASE() local