Merge pull request #20478 from rogday:onnx_refactor_master

This commit is contained in:
Alexander Alekhin 2021-07-30 14:38:45 +00:00
commit 610e3dccb7
3 changed files with 1744 additions and 1542 deletions

View File

@ -14,6 +14,7 @@ Mutex& getInitializationMutex();
void initializeLayerFactory();
namespace detail {
#define CALL_MEMBER_FN(object, ptrToMemFn) ((object).*(ptrToMemFn))
class NotImplemented : public Layer
{

File diff suppressed because it is too large Load Diff

View File

@ -2894,7 +2894,7 @@ void TFImporter::parseNode(const tensorflow::NodeDef& layer)
DispatchMap::const_iterator iter = dispatch.find(type);
if (iter != dispatch.end())
{
((*this).*(iter->second))(net, layer, layerParams);
CALL_MEMBER_FN(*this, iter->second)(net, layer, layerParams);
}
else if (!DNN_DIAGNOSTICS_RUN || !layerHandler->handleMissing(layer))
{