|
|
@@ -42,7 +42,7 @@ FMK_FUNC_HOST_VISIBILITY OpRegistrationTbe *OpRegistrationTbe::Instance() { |
|
|
|
return &instance; |
|
|
|
} |
|
|
|
|
|
|
|
bool OpRegistrationTbe::Finalize(const OpRegistrationData ®_data, bool is_train) onst { |
|
|
|
bool OpRegistrationTbe::Finalize(const OpRegistrationData ®_data, bool is_train) const { |
|
|
|
static std::map<domi::FrameworkType, std::map<std::string, std::string> *> op_map = {{domi::CAFFE, &caffe_op_map}}; |
|
|
|
if (is_train) { |
|
|
|
op_map[domi::TENSORFLOW] = &tensorflow_train_op_map; |
|
|
|