wehub-resource-sync
20 lines
506 B
C++
20 lines
506 B
C++
#include <string.h>
|
|
#include "TfUtils.hpp"
|
|
#include "tfOpConverter.hpp"
|
|
|
|
DECLARE_OP_CONVERTER(SegmentMeanTf);
|
|
|
|
MNN::OpType SegmentMeanTf::opType() {
|
|
return MNN::OpType_Segment;
|
|
}
|
|
MNN::OpParameter SegmentMeanTf::type() {
|
|
return MNN::OpParameter_ReductionParam;
|
|
}
|
|
|
|
void SegmentMeanTf::run(MNN::OpT *dstOp, TmpNode *srcNode) {
|
|
dstOp->main.value = new MNN::ReductionParamT;
|
|
dstOp->main.AsReductionParam()->operation = MNN::ReductionType_MEAN;
|
|
}
|
|
|
|
REGISTER_CONVERTER(SegmentMeanTf, SegmentMean);
|