def __init__(self,
atomic_clean_policy=0,
static_memory_policy=0
)
MemoryConfig类的构造函数,用于配置系统内存使用方式。
无。
参数名 |
输入/输出 |
描述 |
---|---|---|
atomic_clean_policy |
输入 |
是否集中清理网络中所有atomic算子占用的内存,取值包括:
|
static_memory_policy |
输入 |
网络运行时使用的内存分配方式。
默认值是0。
说明:
|
返回MemoryConfig类对象,作为NPURunConfig的参数传入。