MegEngine/dnn/include/megdnn/internal/opr_header_prologue.h

// intentional no header guard here

#include "megdnn/handle.h"
#include "megdnn/opr_param_defs.h"
#include "megdnn/opr_result_defs.h"
#include "megdnn/oprs/base.h"

#include "./visibility_prologue.h"

#include <array>
#include <limits>

#ifndef _megdnn_in
#define _megdnn_in
#endif

#ifndef _megdnn_out
#define _megdnn_out
#endif

#ifndef _megdnn_tensor_in
#define _megdnn_tensor_in const TensorND&
#endif

#ifndef _megdnn_tensor_out
#define _megdnn_tensor_out const TensorND&
#endif

#ifndef _megdnn_tensor_inout
#define _megdnn_tensor_inout const TensorND&
#endif

#ifndef _megdnn_workspace
#define _megdnn_workspace const Workspace&
#endif

#define DEF_OPR_IMPL_CTOR(_opr_name, _base_name) \
public:                                          \
    _opr_name(Handle* handle) : _base_name(handle) {}

#define DEF_OPR_IMPL(_opr_name, _base_name, _nr_inputs, _nr_outputs) \
    DEF_OPR_IMPL_CTOR(_opr_name, _base_name)                         \
    static MEGDNN_CONSTEXPR int NR_INPUTS = _nr_inputs;              \
    static MEGDNN_CONSTEXPR int NR_OUTPUTS = _nr_outputs;

#define DEF_OPR_PARAM(_pname)                      \
public:                                            \
    using Param = param::_pname;                   \
    Param& param() { return m_param; }             \
    const Param& param() const { return m_param; } \
                                                   \
protected:                                         \
    Param m_param

// vim: syntax=cpp.doxygen