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

Method manage

src/runtime/IWeightsManager.cpp:32–53  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

30}
31
32void IWeightsManager::manage(const ITensor *weights, ITransformWeights *parent)
33{
34 if (!are_weights_managed(weights))
35 {
36 _managed_weights[weights];
37 _managed_counter[weights];
38 }
39 else
40 {
41 _managed_counter[weights].counter++;
42 }
43
44 // In case the weights are an output of a previous reshape function
45 // store the parent's link
46 if (parent != nullptr)
47 {
48 if (_managed_weights_parents.find(weights) == _managed_weights_parents.end())
49 {
50 _managed_weights_parents[weights] = parent;
51 }
52 }
53}
54
55ITensor *IWeightsManager::run(const ITensor *weights, ITransformWeights *weights_transform)
56{

Callers 15

manage_workspaceFunction · 0.45
configureMethod · 0.45
configure_layer_normMethod · 0.45
configure_mmMethod · 0.45
configureMethod · 0.45
configureMethod · 0.45
configureMethod · 0.45
configureMethod · 0.45
configureMethod · 0.45
configureMethod · 0.45
configureMethod · 0.45
configureMethod · 0.45

Calls 2

findMethod · 0.80
endMethod · 0.45

Tested by

no test coverage detected