1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/variable_ops.h b/tensorflow/core/kernels/variable_ops.h
index f0b5796d04..355140d44c 100644
--- a/tensorflow/core/kernels/variable_ops.h
+++ b/tensorflow/core/kernels/variable_ops.h
@@ -180,10 +180,10 @@ class DestroyTemporaryVariableOp : public OpKernel {
     if (context->track_allocations()) {
       if (context->allocate_on_host(AllocatorAttributes())) {
         context->record_host_persistent_memory_allocation(
-            -tmpvar.AllocatedBytes());
+            -static_cast<int64>(tmpvar.AllocatedBytes()));
       } else {
         context->record_device_persistent_memory_allocation(
-            -tmpvar.AllocatedBytes());
+            -static_cast<int64>(tmpvar.AllocatedBytes()));
       }
     }
   }