MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / allocate_const_tensors

Function allocate_const_tensors

src/graph/detail/ExecutionHelpers.cpp:99–118  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

97}
98
99void allocate_const_tensors(Graph &g)
100{
101 for (auto &node : g.nodes())
102 {
103 if (node != nullptr)
104 {
105 switch (node->type())
106 {
107 case NodeType::Const:
108 case NodeType::Input:
109 allocate_all_output_tensors(*node);
110 break;
111 case NodeType::Output:
112 allocate_all_input_tensors(*node);
113 default:
114 break;
115 }
116 }
117 }
118}
119
120void allocate_all_tensors(Graph &g)
121{

Callers 1

finalize_graphMethod · 0.85

Calls 3

typeMethod · 0.45

Tested by

no test coverage detected