BackingDeviceName returns the name of the device in whose memory the tensor handle resides. This function will block till the operation that produces `h` has completed. WARNING: The implementation currently returns the same as DeviceName(). After TensoFlow 1.13's C library is released, this impleme
()
| 128 | // After TensoFlow 1.13's C library is released, this implementation will |
| 129 | // be updated to return what the documentation says! |
| 130 | func (th *TensorHandle) BackingDeviceName() (string, error) { |
| 131 | // TODO(ashankar): Restore after TensorFlow 1.13 is released. |
| 132 | // See https://github.com/tensorflow/tensorflow/issues/23257#issuecomment-433751410 |
| 133 | return th.DeviceName() |
| 134 | /* |
| 135 | status := newStatus() |
| 136 | name := C.TFE_TensorHandleBackingDeviceName(th.c, status.c) |
| 137 | if err := status.Err(); err != nil { |
| 138 | return "", err |
| 139 | } |
| 140 | return C.GoString(name), nil |
| 141 | */ |
| 142 | } |
| 143 | |
| 144 | // ToTensor returns the Tensor referenced by th. It may block if this tensor is |
| 145 | // not yet computed. |