serjl issues

Repositories
Issues
Comments

Results 2 issues of


                                            serjl

Creat Gpu TFTensor from Cuda array on GPU to avoid the deviceTohost copy.

**Is your feature request related to a problem? Please describe.** TFTensor object usually obtained from a CPU array meaning a need of data copy from a device(GPU) to a host...

Iteration through graph nodes and set devices

There is a rather simple way to iterate graph nodes and set a device in Java TF: String deviceString = String.format("/gpu:%d", deviceIdx); GraphDef.Builder builder = GraphDef.parseFrom(graphDef).toBuilder(); for (int i =...