enum_macro.h 1.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
// clang-format off
#define FOR_EACH_ENUM_PARAM(cb) \
    cb(::megdnn::param::PoolingV0::Mode); \
    cb(::megdnn::param::Convolution::Format); \
    cb(::megdnn::param::Argsort::Order); \
    cb(::megdnn::param::ConvBiasV0::NonlineMode); \
    cb(::megdnn::param::ConvolutionV0::Mode); \
    cb(::megdnn::param::ConvolutionV0::Sparse); \
    cb(::megdnn::param::ConvolutionV1::ComputeMode); \
    cb(::megdnn::param::BN::ParamDim); \
    cb(::megdnn::param::BN::FwdMode); \
    cb(::megdnn::param::MatrixMulV1::ComputeMode); \
    cb(::megdnn::param::MatrixMul::Format); \
    cb(::megdnn::param::CollectiveComm::Mode); \
    cb(::megdnn::param::Convolution3D::Mode); \
    cb(::megdnn::param::Convolution3D::Sparse); \
    cb(::megdnn::param::Convolution3D::DataType); \
    cb(::megdnn::param::Convolution3D::Format); \
    cb(::megdnn::param::ConvolutionV0::Format); \
    cb(::megdnn::param::CvtColor::Mode); \
    cb(::megdnn::param::Elemwise::Mode); \
    cb(::megdnn::param::ElemwiseMultiType::Mode); \
23 24
    cb(::megdnn::param::MultiHeadAttn::ATTN_MASK_TYPE); \
    cb(::megdnn::param::MultiHeadAttn::TENSOR_COMBINATION_TYPE); \
25 26 27 28 29 30 31 32 33 34 35 36
    cb(::megdnn::param::Padding::PaddingMode); \
    cb(::megdnn::param::RNNCell::NonlineMode); \
    cb(::megdnn::param::ROIAlignV0::Mode); \
    cb(::megdnn::param::ROIPooling::Mode); \
    cb(::megdnn::param::Reduce::Mode); \
    cb(::megdnn::param::Reduce::DataType); \
    cb(::megdnn::param::WarpPerspectiveV1::InterpolationMode); \
    cb(::megdnn::param::WarpPerspectiveV1::BorderMode); \
    cb(::megdnn::param::TopK::Mode);
#define FOR_EACH_BIT_COMBINED_ENUM_PARAM(cb) \
    cb(::megdnn::param::ExecutionPolicy::Strategy);
// clang-format on