| @@ -54,10 +54,11 @@ class CaffeModelParser : public domi::ModelParser { | |||||
| */ | */ | ||||
| Status Parse(const char *file, ge::Graph &graph) override; | Status Parse(const char *file, ge::Graph &graph) override; | ||||
| Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | ||||
| #ifndef ONLY_COMPILE_OPEN_SRC | |||||
| virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | ||||
| return domi::SUCCESS; | return domi::SUCCESS; | ||||
| } | } | ||||
| #endif | |||||
| /** | /** | ||||
| * @ingroup domi_omg | * @ingroup domi_omg | ||||
| * @brief Convert model files to JSON format | * @brief Convert model files to JSON format | ||||
| @@ -39,10 +39,11 @@ class OnnxModelParser : public domi::ModelParser { | |||||
| ge::DataType ConvertToGeDataType(const uint32_t type) override; | ge::DataType ConvertToGeDataType(const uint32_t type) override; | ||||
| Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override { return domi::SUCCESS; } | Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override { return domi::SUCCESS; } | ||||
| #ifndef ONLY_COMPILE_OPEN_SRC | |||||
| virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | ||||
| return domi::SUCCESS; | return domi::SUCCESS; | ||||
| } | } | ||||
| #endif | |||||
| Status ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) override { | Status ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) override { | ||||
| return domi::SUCCESS; | return domi::SUCCESS; | ||||
| } | } | ||||
| @@ -91,10 +91,11 @@ class TensorFlowModelParser : public domi::ModelParser { | |||||
| Status Parse(const char *file, ge::Graph &graph) override; | Status Parse(const char *file, ge::Graph &graph) override; | ||||
| Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | Status ParseFromMemory(const char *data, uint32_t size, ge::ComputeGraphPtr &graph) override; | ||||
| #ifndef ONLY_COMPILE_OPEN_SRC | |||||
| virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | virtual Status ParseFromMemory(const char *data, uint32_t size, ge::Graph &graph) { | ||||
| return domi::SUCCESS; | return domi::SUCCESS; | ||||
| } | } | ||||
| #endif | |||||
| /** | /** | ||||
| * @ingroup domi_omg | * @ingroup domi_omg | ||||
| * @brief Convert model files to JSON format | * @brief Convert model files to JSON format | ||||