device_guard

API属性:声明式编程(静态图)专用API

paddle.fluid. device_guard ( device=None ) [源代码]

一个用于指定OP运行设备的上下文管理器。