Skip to content

Commit

Permalink
add blank line
Browse files Browse the repository at this point in the history
  • Loading branch information
wejoncy committed Nov 8, 2024
1 parent e0b46f7 commit 23b540b
Showing 1 changed file with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -117,6 +117,7 @@ bool SqueezeOpBuilder::IsOpSupportedImpl(const Node& node, const OpBuilderInputP
if (!input_params.create_mlprogram) {
return false;
}

int64_t num_of_dims = 0;
if (node.SinceVersion() > 12) {
num_of_dims = node.InputDefs()[1]->Shape()->dim(0).dim_value();
Expand All @@ -125,6 +126,7 @@ bool SqueezeOpBuilder::IsOpSupportedImpl(const Node& node, const OpBuilderInputP
auto axes = helper.Get("axes", std::vector<int64_t>());
num_of_dims = static_cast<int64_t>(axes.size());
}

std::vector<int64_t> input_shape;
if (!GetShape(*input_defs[0], input_shape, logger) || input_shape.size() + num_of_dims > 5) {
LOGS(logger, VERBOSE) << "Unsqueeze with num_of_dims > 5 is not supported";
Expand Down

0 comments on commit 23b540b

Please sign in to comment.