mxnet

MxNet源码解析(6) registry

2018-09-15  本文已影响0人  Junr_0926

1. 前言

在MXNet中处处都用到了registry,因此有必要了解一下它的实现和用法。

2. registry.h

class Registry:类Registry用于注册全局的单例类。在MXNet中,各种单例类都是通过一个宏来完成注册的。

FunctionRegEntryBase是注册函数的基本类,它提供了几个函数

  1. DMLC_REGISTRY_ENABLE:当我们需要一个单例模式的时候,就要将它使用该宏来初始化,该宏定义了实例化的RegistryGet()方法,方法返回该注册单例的静态变量
  2. DMLC_REGISTRY_REGISTER(EntryType, EntryTypeName, Name):将Name通过函数__REGISTER__函数来进行注册,
上一篇下一篇

猜你喜欢

热点阅读