[circle2circle] Add substitute_expand_dims_to_reshape option (Samsung…

…#14197) Let's add substitute_expand_dims_to_reshape option to circle2circle cli. ONE-DCO-1.0-Signed-off-by: Dayoung Lee <[email protected]>
lemmaa · Oct 11, 2024 · b423ede · b423ede
1 parent f47659f
commit b423ede
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/compiler/circle2circle/src/Circle2Circle.cpp b/compiler/circle2circle/src/Circle2Circle.cpp
@@ -181,6 +181,8 @@ int entry(int argc, char **argv)
              "it only converts weights whose row is a multiple of 16");
   add_switch(arser, "--replace_non_const_fc_with_batch_matmul",
              "Replace FullyConnected with BatchMatMul when its weight is non-constant");
+  add_switch(arser, "--substitute_expand_dims_to_reshape",
+             "This will convert ExpandDims with constant axis to Reshape");
   add_switch(arser, "--substitute_pack_to_reshape",
              "This will convert single input Pack to Reshape");
   add_switch(arser, "--substitute_padv2_to_pad",
@@ -333,6 +335,7 @@ int entry(int argc, char **argv)
   option_str_to_enum["resolve_former_customop"] = Algorithms::ResolveFormerCustomOp;
   option_str_to_enum["shuffle_weight_to_16x1float32"] = Algorithms::ShuffleWeightTo16x1Float32;
   option_str_to_enum["replace_non_const_fc_with_batch_matmul"] = Algorithms::ReplaceNonConstFCWithBatchMatMul;
+  option_str_to_enum["substitute_expand_dims_to_reshape"] = Algorithms::SubstituteExpandDimsToReshape;
   option_str_to_enum["substitute_pack_to_reshape"] = Algorithms::SubstitutePackToReshape;
   option_str_to_enum["substitute_padv2_to_pad"] = Algorithms::SubstitutePadV2ToPad;
   option_str_to_enum["substitute_splitv_to_split"] = Algorithms::SubstituteSplitVToSplit;