diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-22 12:42:59 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-22 12:46:28 -0800 |
| commit | e4532d20973c4c00854492362665317551661c18 (patch) | |
| tree | 398527e29bd30d39237adb4785be5069fdb646fa /tensorflow/contrib/memory_stats | |
| parent | 673641c2d6a27fa97ee05453d671853731a4c602 (diff) | |
Merge changes from github.
PiperOrigin-RevId: 179953488
Diffstat (limited to 'tensorflow/contrib/memory_stats')
| -rw-r--r-- | tensorflow/contrib/memory_stats/kernels/memory_stats_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/memory_stats/kernels/memory_stats_ops.cc b/tensorflow/contrib/memory_stats/kernels/memory_stats_ops.cc index 7e2e96e160..39c0d5af45 100644 --- a/tensorflow/contrib/memory_stats/kernels/memory_stats_ops.cc +++ b/tensorflow/contrib/memory_stats/kernels/memory_stats_ops.cc @@ -59,7 +59,7 @@ REGISTER_KERNEL_BUILDER(Name("BytesInUse").Device(DEVICE_GPU).HostMemory("out"), #ifdef TENSORFLOW_USE_SYCL REGISTER_KERNEL_BUILDER( - Name("BytesInUse").Device(DEVICE_SYCL).HostMemory("out"), MaxBytesInUseOp); + Name("BytesInUse").Device(DEVICE_SYCL).HostMemory("out"), BytesInUseOp); #endif // TENSORFLOW_USE_SYCL // Op that measures the total memory (in bytes) of a device. |