diff options
author | 2018-08-05 21:52:27 -0700 | |
---|---|---|
committer | 2018-08-05 21:56:36 -0700 | |
commit | 02ae1e2e781b8e049d1fc1ab7b52f6ee7edb4423 (patch) | |
tree | 134171731db099fd5efcf600866e7ad8b8fff22d /tensorflow/core/framework/tensor.cc | |
parent | c42013f103ab8f6588cff1a8e59bc1ef81435bcc (diff) |
[tf.data] Add support for copying `Optional` variants to/from GPU.
PiperOrigin-RevId: 207490563
Diffstat (limited to 'tensorflow/core/framework/tensor.cc')
-rw-r--r-- | tensorflow/core/framework/tensor.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/framework/tensor.cc b/tensorflow/core/framework/tensor.cc index 384a42fc11..5f805f6594 100644 --- a/tensorflow/core/framework/tensor.cc +++ b/tensorflow/core/framework/tensor.cc @@ -57,6 +57,10 @@ namespace tensorflow { // Allow Tensors to be stored inside Variants with automatic // encoding/decoding when those Variants are themselves being decoded // in a Tensor's FromProto. +// +// NOTE(mrry): The corresponding "copy function" registrations can be found in +// ../common_runtime/copy_tensor.cc (due to dependencies on other common_runtime +// code). REGISTER_UNARY_VARIANT_DECODE_FUNCTION(Tensor, "tensorflow::Tensor"); namespace { |