norm.h 362 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

#pragma once
#include <iostream>
#include "megdnn/basic_types.h"
#include "megdnn/opr_param_defs.h"

namespace megdnn {
namespace test {
namespace norm {

struct TestArg {
    param::Norm param;
    TensorShape src;
    TestArg(param::Norm param, TensorShape src) : param(param), src(src) {}
};

}  // namespace norm
}  // namespace test
}  // namespace megdnn