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

[POT] Update dtype for BC Parameters (#9468)

Browse Source
This commit is contained in:
Nikita Malinin
2021-12-29 15:11:56 +03:00
committed by GitHub
parent 1ee8007764
commit 010877d06c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -261,11 +261,12 @@ class BiasCorrection(Algorithm):
inputs_data = []
for input_node in input_nodes:
input_node_name = nu.create_node_name(input_node)
input_node_data_type = nu.get_node_data_type(input_node)
c_input_shape = outputs_shapes[input_node_name]
c_input_shape[0] = 1
parameter_name = input_node_name + '/parameter'
param_node = ge.create_node(input_node.graph, parameter_name, 'Parameter',
{'shape': c_input_shape})
{'shape': c_input_shape, 'data_type': input_node_data_type})
for _, port in input_node.out_ports().items():
for in_port in port.get_destinations():
in_port.disconnect()