Searched refs:m_Graph (Results 1 – 8 of 8) sorted by relevance
| /external/armnn/src/armnn/ |
| D | Graph.hpp | 63 explicit InputLayersAccessor(const Graph& graph) : m_Graph(graph) {} in InputLayersAccessor()
67 return { m_Graph.m_Layers.begin(), &(PtrCast<const InputLayer>) }; in begin()
72 …return { std::next(m_Graph.m_Layers.begin(), static_cast<IteratorDifference>(m_Graph.GetNumInputs(… in end()
76 const Graph& m_Graph; member
82 explicit OutputLayersAccessor(const Graph& graph) : m_Graph(graph) {} in OutputLayersAccessor()
86 …return { std::prev(m_Graph.m_Layers.end(), static_cast<IteratorDifference>(m_Graph.GetNumOutputs()… in begin()
92 return { m_Graph.m_Layers.end(), &(PtrCast<const OutputLayer>) }; in end()
95 const Graph& m_Graph; member
310 : LayerT(std::forward<Args>(args)...), m_Graph(&graph) in LayerInGraphBase()
312 Insert(*m_Graph, insertBefore); in LayerInGraphBase()
[all …]
|
| D | Network.cpp | 713 m_Graph->Print(); in PrintGraph()
719 return m_Graph->SerializeToDot(stream); in SerializeToDot()
724 return m_Graph->GetNumInputs(); in GetNumInputs()
729 return m_Graph->GetNumOutputs(); in GetNumOutputs()
2044 m_Graph(std::make_unique<Graph>(GetShapeInferenceMethod(), GetAllowExpandedDims())) in NetworkImpl()
2053 m_Graph->Print(); in PrintGraph()
2059 return m_Graph->AddLayer<InputLayer>(id, name); in AddInputLayer()
2065 return m_Graph->AddLayer<BatchToSpaceNdLayer>(batchToSpaceNdDescriptor, name); in AddBatchToSpaceNdLayer()
2070 return m_Graph->AddLayer<CastLayer>(name); in AddCastLayer()
2075 return m_Graph->AddLayer<ChannelShuffleLayer>(channelShuffleDescriptor, name); in AddChannelShuffleLayer()
[all …]
|
| D | OptimizedNetworkImpl.hpp | 27 Graph& GetGraph() { return *m_Graph; } in GetGraph()
28 Graph& GetGraph() const { return *m_Graph; } in GetGraph()
34 std::unique_ptr<Graph> m_Graph; member in armnn::OptimizedNetworkImpl
|
| D | SubgraphView.cpp | 420 : m_Graph(graph) in SubgraphViewWorkingCopy()
424 Graph m_Graph; member
446 Layer* const layer = PolymorphicDowncast<const Layer*>(originalLayer)->Clone(ptr->m_Graph); in GetWorkingCopy()
603 auto workingCopyGraph = &p_WorkingCopyImpl->m_Graph; in SubstituteSubgraph()
|
| D | Network.hpp | 39 { return *m_Graph; } in GetGraph()
245 std::unique_ptr<Graph> m_Graph; member in armnn::NetworkImpl
|
| /external/armnn/src/armnn/test/ |
| D | GraphTests.cpp | 496 armnn::Graph m_Graph; member
505 LayerType* const layer = m_Graph.AddLayer<LayerType>(std::forward<Args>(args)...); in AddLayer()
519 const armnn::Graph origGraph(m_Graph);
520 m_Graph.AddCompatibilityLayers(m_Backends, m_FactoryRegistry);
522 TestGraphAfterAddingCopyLayers(m_Graph, origGraph);
528 m_Graph.AddCompatibilityLayers(m_Backends, m_FactoryRegistry);
531 const std::vector<Edge> edges = GetEdgeList(m_Graph);
534 m_Graph.AddCompatibilityLayers(m_Backends, m_FactoryRegistry);
535 const std::vector<Edge> otherEdges = GetEdgeList(m_Graph);
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 562 m_Graph = nullptr; in ResetParser()
868 m_Graph = std::make_unique<onnx::GraphProto>(*model.mutable_graph()); in CreateNetworkFromModel()
882 ARMNN_ASSERT(m_Graph.get() != nullptr); in LoadGraph()
885 SetupInfo(m_Graph->mutable_output()); in LoadGraph()
886 SetupInfo(m_Graph->mutable_input()); in LoadGraph()
887 SetupInfo(m_Graph->mutable_value_info()); in LoadGraph()
889 for (auto tensor : m_Graph->initializer()) in LoadGraph()
904 for(size_t nodeIndex = 0; nodeIndex < static_cast<size_t>(m_Graph->node_size()); nodeIndex++) in LoadGraph()
906 auto node = m_Graph->node(static_cast<int>(nodeIndex)); in LoadGraph()
921 AddFullyConnected(m_Graph->node(matmulIndex), &node); in LoadGraph()
[all …]
|
| D | OnnxParser.hpp | 180 GraphPtr m_Graph; member in armnnOnnxParser::OnnxParserImpl
|