wehub-resource-sync
62 lines
2.2 KiB
C++
62 lines
2.2 KiB
C++
//
|
|
// SpaceToDepthTflite.cpp
|
|
// MNNConverter
|
|
//
|
|
// Created by MNN on 2020/01/02.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include <stdio.h>
|
|
#include "liteOpConverter.hpp"
|
|
using namespace tflite;
|
|
|
|
DECLARE_OP_COVERTER(SpaceToDepthTflite);
|
|
|
|
MNN::OpType SpaceToDepthTflite::opType(int quantizedModel) {
|
|
return MNN::OpType_SpaceToDepth;
|
|
}
|
|
MNN::OpParameter SpaceToDepthTflite::type(int quantizedModel) {
|
|
return MNN::OpParameter_DepthSpaceParam;
|
|
}
|
|
|
|
void SpaceToDepthTflite::run(MNN::OpT* dstOp, const std::unique_ptr<tflite::OperatorT>& tfliteOp,
|
|
const std::vector<std::unique_ptr<tflite::TensorT>>& tfliteTensors,
|
|
const std::vector<std::unique_ptr<tflite::BufferT>>& tfliteModelBuffer,
|
|
const std::vector<std::unique_ptr<tflite::OperatorCodeT>>& tfliteOpSet, int quantizedModel) {
|
|
auto spaceToDepthParam = new MNN::DepthSpaceParamT;
|
|
auto opt=tfliteOp->builtin_options.AsSpaceToDepthOptions();
|
|
spaceToDepthParam->blockSize = opt->block_size;
|
|
dstOp->main.value = spaceToDepthParam;
|
|
|
|
}
|
|
|
|
REGISTER_CONVERTER(SpaceToDepthTflite, BuiltinOperator_SPACE_TO_DEPTH);
|
|
|
|
|
|
/**
|
|
From https://github.com/alibaba/MNN/issues/1452
|
|
Author: https://github.com/pkjq11
|
|
*/
|
|
|
|
DECLARE_OP_COVERTER(DepthToSpaceTflite);
|
|
|
|
MNN::OpType DepthToSpaceTflite::opType(int quantizedModel) {
|
|
return MNN::OpType_DepthToSpace;
|
|
}
|
|
MNN::OpParameter DepthToSpaceTflite::type(int quantizedModel) {
|
|
return MNN::OpParameter_DepthSpaceParam;
|
|
}
|
|
|
|
void DepthToSpaceTflite::run(MNN::OpT* dstOp, const std::unique_ptr<tflite::OperatorT>& tfliteOp,
|
|
const std::vector<std::unique_ptr<tflite::TensorT>>& tfliteTensors,
|
|
const std::vector<std::unique_ptr<tflite::BufferT>>& tfliteModelBuffer,
|
|
const std::vector<std::unique_ptr<tflite::OperatorCodeT>>& tfliteOpSet, int quantizedModel) {
|
|
auto depthToSpaceParam = new MNN::DepthSpaceParamT;
|
|
auto opt=tfliteOp->builtin_options.AsDepthToSpaceOptions();
|
|
depthToSpaceParam->blockSize = opt->block_size;
|
|
dstOp->main.value = depthToSpaceParam;
|
|
|
|
}
|
|
|
|
REGISTER_CONVERTER(DepthToSpaceTflite, BuiltinOperator_DEPTH_TO_SPACE);
|