Skip to content

Commit

Permalink
Fix Conv skipping inputs
Browse files Browse the repository at this point in the history
  • Loading branch information
skottmckay committed Jul 22, 2024
1 parent bb00eae commit 661dc50
Showing 1 changed file with 0 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -28,11 +28,6 @@ class ConvOpBuilder : public BaseOpBuilder {
};

void ConvOpBuilder::AddInitializersToSkip(ModelBuilder& model_builder, const Node& node) const {
if (model_builder.CreateMLProgram()) {
// we add the initializers as 'const' operations via ModelBuilder::RegisterInitializers
return;
}

const auto& input_defs = node.InputDefs();

// skip the weight and bias (if has it) for conv as we will directly set those as part of the NN layer
Expand Down

0 comments on commit 661dc50

Please sign in to comment.