| /external/armnn/src/armnn/test/ |
| D | NetworkTests.cpp | 159 CHECK(net.GetGraph().GetNumLayers() == 13);
162 CHECK(GraphHasNamedLayer(net.GetGraph(), "input layer"));
163 CHECK(GraphHasNamedLayer(net.GetGraph(), "conv layer"));
164 CHECK(GraphHasNamedLayer(net.GetGraph(), "conv const weights"));
165 CHECK(GraphHasNamedLayer(net.GetGraph(), "fc const weights"));
166 CHECK(GraphHasNamedLayer(net.GetGraph(), "fully connected"));
167 CHECK(GraphHasNamedLayer(net.GetGraph(), "pooling2d"));
168 CHECK(GraphHasNamedLayer(net.GetGraph(), "activation"));
169 CHECK(GraphHasNamedLayer(net.GetGraph(), "normalization"));
170 CHECK(GraphHasNamedLayer(net.GetGraph(), "softmax"));
[all …]
|
| D | OptimizerTests.cpp | 820 Graph& optGraph = optNet.GetGraph();
|
| /external/armnn/src/armnn/ |
| D | OptimizedNetworkImpl.hpp | 27 Graph& GetGraph() { return *m_Graph; } in GetGraph() function in armnn::OptimizedNetworkImpl
28 Graph& GetGraph() const { return *m_Graph; } in GetGraph() function in armnn::OptimizedNetworkImpl
|
| D | LoadedNetwork.cpp | 234 Graph& order = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph(); in LoadedNetwork()
583 m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().AllocateDynamicBuffers(); in LoadedNetwork()
627 Graph& order = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph(); in AllocateAndExecuteConstantWorkloadsAsync()
657 Graph& order = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().TopologicalSort(); in SendNetworkStructure()
699 for (auto&& inputLayer : m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().GetInputLayers()) in GetInputTensorInfo()
713 … for (auto&& outputLayer : m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().GetOutputLayers()) in GetOutputTensorInfo()
844 const Graph& graph = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph(); in EnqueueWorkload()
1429 … if (inputTensors.size() > m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().GetNumInputs()) in ImportInputs()
1435 Graph& graph = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().TopologicalSort(); in ImportInputs()
1488 Graph& graph = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().TopologicalSort(); in ImportInputs()
[all …]
|
| D | Network.cpp | 693 return pOptimizedNetworkImpl->GetGraph().GetProfiler(); in GetProfiler()
1022 optNetObjPtr->GetGraph(), in AssignBackendsIConnectable()
1046 optNetObjPtr->GetGraph(), in AssignBackendsIConnectable()
1251 Graph& optGraph = optNetObjPtr->GetGraph(); in ApplyBackendOptimizations()
1835 Graph& optGraph = optNetObjPtr->pOptimizedNetworkImpl->GetGraph(); in Optimize()
2007 return Optimize(inNetwork.pNetworkImpl->GetGraph(), in Optimize()
2938 for (auto layer : GetGraph()) in ExecuteStrategy()
2972 for (auto layer : GetGraph()) in ExecuteStrategy()
|
| D | Network.hpp | 38 const Graph& GetGraph() const in GetGraph() function in armnn::NetworkImpl
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/ |
| D | gen_tftrt_model.py | 45 def GetGraph(input1, input2, var): function
71 return GetGraph(input1, input2, self.v)
106 out = GetGraph(input1, input2, var)
|
| /external/armnn/src/armnnTestUtils/ |
| D | TestUtils.cpp | 51 return optNet->pOptimizedNetworkImpl->GetGraph(); in GetGraphForTesting()
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp | 90 Graph graph = network.get()->GetGraph();
120 Graph graph = network.get()->GetGraph();
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 177 Graph* GetGraph() const;
447 Graph* Encapsulator::Subgraph::GetGraph() const { return graph_.get(); } in GetGraph() function in tensorflow::__anon018d25a70111::Encapsulator::Subgraph
694 Graph* g = subgraphs_[src_func_id].GetGraph(); in CopySubgraphEdges()
769 FixupSourceAndSinkEdges(subgraph.GetGraph()); in SplitIntoSubgraphs()
777 *entry.second.GetGraph(), library); in SplitIntoSubgraphs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 311 std::unique_ptr<Graph> GetGraph(const FunctionBody* fbody);
|
| D | graph_compiler.cc | 255 auto graph = compiler->GetGraph(fbody); in CompileFunctionalNode()
|
| D | xla_compiler.cc | 584 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph() function in tensorflow::XlaCompiler
788 std::unique_ptr<Graph> graph = GetGraph(fbody); in CompileFunction()
|
| /external/armnn/src/backends/cl/test/ |
| D | ClImportTensorHandleTests.cpp | 480 …IOptimizedNetworkPtr optNet = Optimize(network.GetGraph(), backends, runtime->GetDeviceSpec(), opt…
628 …IOptimizedNetworkPtr optNet = Optimize(network.GetGraph(), backends, runtime->GetDeviceSpec(), opt…
767 …IOptimizedNetworkPtr optNet = Optimize(network.GetGraph(), backends, runtime->GetDeviceSpec(), opt…
|