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 QuantizeDequantizeWeights,
59 ForwardReshapeToUnaryOp,
61 FusePreActivationBatchNorm,
62 MakeBatchNormGammaPositive,
63 FuseActivationFunction,
64 ShuffleWeightTo16x1Float32,
65 RemoveRedundantTranspose,
66 ReplaceMulAddWithDepthwiseConv,
68 SubstitutePackToReshape,
70 SubstituteSplitVToSplit,
71 SubstituteSqueezeToReshape,
74 RemoveUnnecessarySlice,
75 RemoveUnnecessaryStridedSlice,
76 RemoveUnnecessarySplit,
77 RemoveUnnecessaryReshape,
78 TransformMinMaxToRelu6Pass,
79 TransformMinReluToRelu6Pass,
80 SubstituteStridedSliceToReshape,
81 SubstituteTransposeToReshape,
82 RemoveRedundantReshape,
84 RemoveQuantDequantSeq,
87 enum AlgorithmParameters
90 Quantize_input_model_dtype,
91 Quantize_output_model_dtype,
92 Quantize_granularity, // layer-wise or channel-wise
93 Quantize_tensor_names,
99 Sparsify_traversal_order,
104 // convert NCHW to NHWC
105 NCHW_to_NHWC_input_shape,
106 NCHW_to_NHWC_output_shape,
108 Quantize_input_dtype = Quantize_input_model_dtype, // TODO Remove this
109 Quantize_output_dtype = Quantize_output_model_dtype, // TODO Remove this
112 virtual ~Options() = default;
114 virtual void enable(Algorithm) = 0;
115 virtual bool query(Algorithm) = 0;
116 virtual void param(AlgorithmParameters, const std::string &) = 0;
117 virtual const std::string param(AlgorithmParameters) const = 0;
118 virtual void params(AlgorithmParameters, std::vector<std::string> &) = 0;
119 virtual std::vector<std::string> params(AlgorithmParameters) const = 0;
123 // TODO maybe caller can provide Options as ctor parameters
124 Options *options(void);
127 void optimize(luci::Module *) const;
129 void optimize(loco::Graph *) const;
131 void quantize(loco::Graph *) const;
133 void sparsify(loco::Graph *) const;
136 std::unique_ptr<Options> _options;
141 #endif // __LUCI_CIRCLE_OPTIMIZER_H__