diff options
| author | Alexey Suhov <asuhov@users.noreply.github.com> | 2019-01-21 21:31:31 +0300 |
|---|---|---|
| committer | openvino-pushbot <44090433+openvino-pushbot@users.noreply.github.com> | 2019-01-21 21:31:31 +0300 |
| commit | 9de27f16bc8b712a5b8c99d1d4b4a66c9144942d (patch) | |
| tree | 01a383efe94d92b9870d513c2c5ea5d15b07010a /model-optimizer/extensions/middle/ConvertGroupedStridedSlice.py | |
| parent | fbc7a4a710c24def8ab199926a7da90a0394b87d (diff) | |
| download | dldt-9de27f16bc8b712a5b8c99d1d4b4a66c9144942d.tar.gz dldt-9de27f16bc8b712a5b8c99d1d4b4a66c9144942d.tar.bz2 dldt-9de27f16bc8b712a5b8c99d1d4b4a66c9144942d.zip | |
Publishing R5 content (#72)
* Publishing R5 content
* Updated ade revision
* updated readme
* add possibility to build CPU plugin with Intel MKL package
Diffstat (limited to 'model-optimizer/extensions/middle/ConvertGroupedStridedSlice.py')
| -rw-r--r-- | model-optimizer/extensions/middle/ConvertGroupedStridedSlice.py | 16 |
1 files changed, 3 insertions, 13 deletions
diff --git a/model-optimizer/extensions/middle/ConvertGroupedStridedSlice.py b/model-optimizer/extensions/middle/ConvertGroupedStridedSlice.py index 2f5fa88c4..cec09ccb7 100644 --- a/model-optimizer/extensions/middle/ConvertGroupedStridedSlice.py +++ b/model-optimizer/extensions/middle/ConvertGroupedStridedSlice.py @@ -25,7 +25,6 @@ from mo.ops.reshape import Reshape from mo.middle.replacement import MiddleReplacementPattern from extensions.middle.SliceConverter import ConvertSlice - class ConvertGroupedStridedSlice(MiddleReplacementPattern): """ This pass converts subgraphs where StridedSlices used for splitting single channel to single Split layers @@ -110,15 +109,10 @@ class ConvertGroupedStridedSlice(MiddleReplacementPattern): shape = np.array(input_shape) size_splits.append(l - prev_r) shape[split_channel_dim] = l - prev_r - data_node = Op._create_data_node(graph, 'fake_data', {'shape': shape}) - # added fake Reshape to workaround IE issue with Split and fake nodes - fake_op = Reshape(graph, dict(name=out_nodes[0].name + "/" + str(l) + "_fake_op", dim=shape[1:])) - fake_out_node = Op._create_data_node(graph, 'fake_out_data', - {'shape': shape[1:], 'is_output': True}) - fake_op.create_node_with_data([data_node], fake_op.attrs, data_nodes=[fake_out_node]) - + data_node = Op._create_data_node(graph, 'fake_data', {'shape': shape, 'is_output': True}) final_data_nodes_list.append(data_node) + prev_r = r size_splits.append(r - l) final_data_nodes_list.append(out) @@ -130,11 +124,7 @@ class ConvertGroupedStridedSlice(MiddleReplacementPattern): shape = input_shape.copy() shape[split_channel_dim] = input_shape[split_channel_dim] - prev_r size_splits.append(input_shape[split_channel_dim] - prev_r) - data_node = Op._create_data_node(graph, 'fake_data', {'shape': shape}) - # added fake Reshape to workaround IE issue with Split and fake nodes - fake_op = Reshape(graph, dict(name=out_nodes[0].name + "/" + str(l) + "_fake_op", dim=shape[1:])) - fake_out_node = Op._create_data_node(graph, 'fake_out_data', {'shape': shape[1:], 'is_output': True}) - fake_op.create_node_with_data([data_node], fake_op.attrs, data_nodes=[fake_out_node]) + data_node = Op._create_data_node(graph, 'fake_data', {'shape': shape, 'is_output': True}) final_data_nodes_list.append(data_node) if not valid_for_replacement: |