|
@@ -58,7 +58,7 @@ class TBEPluginManager { |
|
|
const string &caffe_parser_so_suff, const string &aicpu_so_suff, |
|
|
const string &caffe_parser_so_suff, const string &aicpu_so_suff, |
|
|
const string &aicpu_host_so_suff); |
|
|
const string &aicpu_host_so_suff); |
|
|
static void FindParserSo(const string &path, vector<string> &file_list, string &caffe_parser_path, |
|
|
static void FindParserSo(const string &path, vector<string> &file_list, string &caffe_parser_path, |
|
|
int32_t recursive_depth = 0); |
|
|
|
|
|
|
|
|
int recursive_depth = 0); |
|
|
static void GetPluginSoFileList(const string &path, vector<string> &file_list, string &caffe_parser_path); |
|
|
static void GetPluginSoFileList(const string &path, vector<string> &file_list, string &caffe_parser_path); |
|
|
static void GetCustomOpPath(std::string &customop_path); |
|
|
static void GetCustomOpPath(std::string &customop_path); |
|
|
void LoadCustomOpLib(); |
|
|
void LoadCustomOpLib(); |
|
|