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

[frontend][paddle] It is possible that get_output_partial_shape returns dynamic shape. Then we cannot simply check if it is 1. In case of BERT which input shape is dynamic, this check will absolutely fail. (#8030)

Browse Source
This commit is contained in:
cecilia peng 2021-10-25 20:36:13 +08:00 committed by GitHub
parent 57a7d3dfcf
commit 1858fa3853
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ngraph/frontend/paddlepaddle/src/op/slice.cpp
View File

@ -77,13 +77,6 @@ NamedOutputs slice(const NodeContext& node) {
auto decrease_axis = node.get_attribute<std::vector<int32_t>>("decrease_axis");
if (decrease_axis.size() > 0) {
auto stride_slice_output_shape = stride_slice_node->get_output_partial_shape(0);
for (size_t i = 0; i < decrease_axis.size(); ++i)
PDPD_OP_VALIDATION_CHECK(node,
stride_slice_output_shape[decrease_axis[i]] == 1,
"decrease dim should be 1!");
auto squeeze_index_node = Constant::create(element::i32, {}, decrease_axis);
auto decreased_node = std::make_shared<Squeeze>(stride_slice_node, squeeze_index_node);