mirror of https://github.com/alibaba/MNN.git
34 lines
822 B
C++
34 lines
822 B
C++
//
|
|
// winogradExample.cpp
|
|
// MNN
|
|
//
|
|
// Created by MNN on 2019/01/22.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <MNN/MNNDefine.h>
|
|
#include "math/Matrix.hpp"
|
|
#include "math/WingoradGenerater.hpp"
|
|
|
|
int main(int argc, const char* argv[]) {
|
|
int unit = ::atoi(argv[1]);
|
|
int kernelSize = ::atoi(argv[2]);
|
|
float interp = 0.5f;
|
|
if (argc > 3) {
|
|
interp = ::atof(argv[3]);
|
|
}
|
|
MNN::Math::WinogradGenerater generater(unit, kernelSize, interp);
|
|
auto a = generater.A();
|
|
auto b = generater.B();
|
|
auto g = generater.G();
|
|
MNN_PRINT("A=\n");
|
|
MNN::Math::Matrix::print(a.get());
|
|
MNN_PRINT("B=\n");
|
|
MNN::Math::Matrix::print(b.get());
|
|
MNN_PRINT("G=\n");
|
|
MNN::Math::Matrix::print(g.get());
|
|
return 0;
|
|
}
|