diff options
Diffstat (limited to 'model-optimizer/mo/ops/split.py')
-rw-r--r-- | model-optimizer/mo/ops/split.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-optimizer/mo/ops/split.py b/model-optimizer/mo/ops/split.py index 145680b46..5ce6b0f65 100644 --- a/model-optimizer/mo/ops/split.py +++ b/model-optimizer/mo/ops/split.py @@ -42,7 +42,7 @@ class Split(Op): input_node = node.in_node(0) outputs = node.out_nodes() out_shape = copy.copy(input_node.shape) - out_shape[node.axis] = np.int64(input_node.shape[node.axis] / node.pb.num_split) + out_shape[node.axis] = np.int64(input_node.shape[node.axis] / node.num_split) for idx, output in outputs.items(): output.shape = out_shape PermuteAttrs.create_permute_attrs(node, attrs=[('axis', 'input:0')]) |