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

Fix FBC missed spaces (#11091)

Browse Source
This commit is contained in:
Nikita Malinin
2022-03-23 11:17:42 +03:00
committed by GitHub
parent 99f0093615
commit eb74afe417
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/pot/openvino/tools/pot/algorithms/quantization/fast_bias_correction/algorithm.py
View File

@@ -159,8 +159,8 @@ class FastBiasCorrection(Algorithm):
bias = nu.get_bias_for_node(op_node)
after_biased_conv = nu.get_node_output(bias, 0)[0]
after_biased_conv_name = get_quantized_input_key(after_biased_conv)
inputs_outputs_layout[after_biased_conv_name] = \
{"shape": TensorStatistic(func=lambda x, **kwargs: x.shape, shape_for_inference=True)}
inputs_outputs_layout[after_biased_conv_name] = \
{"shape": TensorStatistic(func=lambda x, **kwargs: x.shape, shape_for_inference=True)}
return inputs_outputs_layout