Commit Graph

4 Commits (7c42f056e2ece4080ac1203489b9730ceb3837eb)

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