IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[POT] Update add node name (#8487)

Browse Source 
* Update node name to make ports visible

* proxy algo

* m

* mistype for add node name

* undo_utils

* Update utils.py

* Update algorithm.py

* undo
This commit is contained in:
Milana Shhanukova 2021-11-11 10:35:00 +03:00 committed by GitHub
parent 0343babacb
commit 700beb7d1f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/pot/openvino/tools/pot/algorithms/quantization/bias_correction/algorithm.py
View File

@ -409,7 +409,7 @@ class BiasCorrection(Algorithm):
input_name = biased_after_param_nodes[node.name]
statistics_layout[input_name] = {'batch_mean_param_in': agf.batch_mean}
self._collected_stat_inputs.append(input_name)
statistics_layout[add_node.name] =\
statistics_layout[add_node_name] =\
{'mean_per_channel': TensorStatisticAxis(granularity='perchannel',
type='mean',
inplace_statistics=self.config['inplace_statistics'],