2 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #ifndef __LUCI_CIRCLE_OPTIMIZER_H__
18 #define __LUCI_CIRCLE_OPTIMIZER_H__
22 #include <luci/IR/Module.h>
30 class CircleOptimizer final
37 FuseAddWithFullyConnected,
39 FuseBatchNormWithConv,
40 FuseBatchNormWithDwConv,
41 FuseBatchNormWithTConv,
45 FuseTransposeWithMean,
47 ResolveCustomOpBatchMatMul,
48 ResolveCustomOpMatMul,
49 ResolveCustomOpMaxPoolWithArgmax,
50 ResolveCustomOpSplitV,
59 ForwardReshapeToUnaryOp,
62 FusePreActivationBatchNorm,
63 MakeBatchNormGammaPositive,
64 FuseActivationFunction,
67 ShuffleWeightTo16x1Float32,
68 RemoveRedundantTranspose,
69 ReplaceMulAddWithDepthwiseConv,
70 ReplaceNonConstFCWithBatchMatMul,
72 SubstitutePackToReshape,
74 SubstituteSplitVToSplit,
75 SubstituteSqueezeToReshape,
78 RemoveUnnecessarySlice,
79 RemoveUnnecessaryStridedSlice,
80 RemoveUnnecessarySplit,
81 RemoveUnnecessaryReshape,
82 TransformMinMaxToRelu6Pass,
83 TransformMinReluToRelu6Pass,
84 DecomposeHardSwishPass,
85 SubstituteStridedSliceToReshape,
86 SubstituteTransposeToReshape,
87 RemoveRedundantQuantize,
88 RemoveRedundantReshape,
90 RemoveQuantDequantSeq,
95 enum AlgorithmParameters
99 Sparsify_traversal_order,
104 // convert NCHW to NHWC
105 NCHW_to_NHWC_input_shape,
106 NCHW_to_NHWC_output_shape,
109 virtual ~Options() = default;
111 virtual void enable(Algorithm) = 0;
112 virtual bool query(Algorithm) = 0;
113 virtual void param(AlgorithmParameters, const std::string &) = 0;
114 virtual const std::string param(AlgorithmParameters) const = 0;
118 // TODO maybe caller can provide Options as ctor parameters
119 Options *options(void);
122 void optimize(luci::Module *) const;
124 void optimize(loco::Graph *) const;
126 void sparsify(loco::Graph *) const;
129 std::unique_ptr<Options> _options;
134 #endif // __LUCI_CIRCLE_OPTIMIZER_H__