Commit 5c41a7ce authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

DirectSession: Add accessor to DeviceMgr. 

This will enable some experimentation with keeping TF_Tensor objects
in the C API backed by device memory (avoiding copies to host memory).

PiperOrigin-RevId: 163173902
parent b01346de
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment