Skip to content

Commit ba03af9

Browse files
Pull request pytorch#88: Turn off optimization REPLACE_AVERAGE_POOL_BEFORE_FULLY_CONNECTED_WITH_SUM
Merge in AITEC/executorch from feature/nxg11066/EIEX-194-turn-off-optimization-ReplaceAveragePoolBeforeFullyConnectedWithSum to main-nxp * commit '5ef87a69f050c257230779421e87eddd666cd8ec': Remove optimization REPLACE_AVERAGE_POOL_BEFORE_FULLY_CONNECTED_WITH_SUM
2 parents 686859f + 5ef87a6 commit ba03af9

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

backends/nxp/backend/ir/tflite_optimizer/optimizer.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,6 @@
3232
RemoveIdentityTransposeOperators
3333
from executorch.backends.nxp.backend.ir.tflite_optimizer.optimizations.remove_unused_tensors_and_buffers import \
3434
RemoveUnusedTensorsAndBuffers
35-
from executorch.backends.nxp.backend.ir.tflite_optimizer.optimizations.replace_average_pool_before_fully_connected_with_sum import \
36-
ReplaceAveragePoolBeforeFullyConnectedWithSum
3735

3836

3937
class Optimization(Enum):
@@ -60,7 +58,6 @@ class Optimization(Enum):
6058

6159
MOVE_ACTIVATION_BEFORE_CONCAT = 15
6260
COMBINE_HARD_SIGMOID_AND_MUL_INTO_HARD_SWISH = 16
63-
REPLACE_AVERAGE_POOL_BEFORE_FULLY_CONNECTED_WITH_SUM = 17
6461

6562

6663
class Optimizer:
@@ -111,9 +108,7 @@ def __init__(self, builder: 'model_builder.ModelBuilder', conversion_config: Con
111108

112109
Optimization.MOVE_ACTIVATION_BEFORE_CONCAT: MoveActivationBeforeConcatenation(builder, conversion_config),
113110
Optimization.COMBINE_HARD_SIGMOID_AND_MUL_INTO_HARD_SWISH:
114-
CombineHardSigmoidAndMulIntoHardSwish(builder, conversion_config),
115-
Optimization.REPLACE_AVERAGE_POOL_BEFORE_FULLY_CONNECTED_WITH_SUM:
116-
ReplaceAveragePoolBeforeFullyConnectedWithSum(builder, conversion_config)
111+
CombineHardSigmoidAndMulIntoHardSwish(builder, conversion_config)
117112
}
118113

119114
def optimize(self, optimization_whitelist: list[Optimization] | None = None,

0 commit comments

Comments
 (0)