op_def_api.h.in 261 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
// the folder of pbtxt with op attribute definition
#pragma once

#include "paddle/fluid/framework/op_def.pb.h"

#define OP_DEF_FOLDER "@OP_DEF_FOLDER@"

namespace paddle {
namespace framework {
    const proto::OpDef& GetOpDef(const std::string& op_name);
}
}