Commit Graph

4 Commits (e657d7062d2c6f5470a505696bbff507839f6e84)

Author SHA1 Message Date
lilong12 29de0d97a5
add the support to specify device index for device_guard (#24555)
5 years ago
Zhang Ting 3f371db82a
add op_device attr for backward op_desc, test=develop (#23062)
5 years ago
Zhang Ting eec10aaba2
set op_device for loss_op_desc (#23027)
5 years ago
Zhang Ting 4e8bc02461
add fluid.device_guard to specify the device type for Op (#22254)
5 years ago