首页 > Web开发 > 详细

mxnet 源码阅读 8

时间:2017-08-13 19:10:32      阅读:436      评论:0      收藏:0      [点我收藏+]

mxnet/include/mxnet/engine.h

在名字空间 mxnet engine 中定义了抽象类Engine,用以规范接口,其接口有

NotifyShutdown

NewVariable

DeleteVariable

NewOperator

DeleteOperator

Push

PushAsync

PushSync

WaitForVar

WaitForAll

DeduplicateVarHandle

mxnet/src/engine/engine_impl.h

定义了多态实现的

Engine *CreateNaiveEngine

Engine *CreateThreadedEnginePolled

Engine *CreateThreadedEnginePerDevice

 mxnet/src/engine/engine.cc

根据环境变量决定Engine *CreateEngine的选择情况。有三种形式

CreateNaiveEngine,CreateThreadedEnginePooled,CreateThreadEnginePerDevice。

 

mxnet 源码阅读 8

原文:http://www.cnblogs.com/reedlau/p/7354285.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!