diff --git a/tools/modelwriter.h b/tools/modelwriter.h index 4f84c736365..8588ee4752a 100644 --- a/tools/modelwriter.h +++ b/tools/modelwriter.h @@ -101,6 +101,7 @@ #include "layer/split.h" #include "layer/squeeze.h" #include "layer/threshold.h" +#include "layer/tile.h" #include "layer/unaryop.h" #include "layer/unfold.h" #include "layer/yolodetectionoutput.h" @@ -2494,6 +2495,17 @@ int ModelWriter::save(const char* parampath, const char* binpath) fprintf_param_value(" 0=%e", threshold) } + else if (layer->type == "Tile") + { + ncnn::Tile* op = (ncnn::Tile*)layer; + ncnn::Tile* op_default = (ncnn::Tile*)layer_default; + + fprintf_param_value(" 0=%d", axis) + fprintf_param_value(" 1=%d", tiles) + { + if (!op->repeats.empty()) fprintf_param_int_array(2, op->repeats, pp); + } + } else if (layer->type == "UnaryOp") { ncnn::UnaryOp* op = (ncnn::UnaryOp*)layer;