Fix pointer error to not take address of existing pointer in TPU stream executor code
PiperOrigin-RevId: 320078141
Change-Id: Ic92ec88e3ce6462b12fe2d6b6edcabeeecc02525
diff --git a/tensorflow/stream_executor/tpu/tpu_platform.cc b/tensorflow/stream_executor/tpu/tpu_platform.cc
index 7a789d0..d0340b1 100644
--- a/tensorflow/stream_executor/tpu/tpu_platform.cc
+++ b/tensorflow/stream_executor/tpu/tpu_platform.cc
@@ -129,7 +129,7 @@
Status TpuPlatform::TpuMemoryLimit(int64* memory_limit) {
TF_Status* status = TF_NewStatus();
tpu::ConfigApiFn()->TpuConfigurationApi_TpuMemoryLimitFn(
- reinterpret_cast<int64_t*>(&memory_limit), status);
+ reinterpret_cast<int64_t*>(memory_limit), status);
auto ret_status = StatusFromTF_Status(status);
TF_DeleteStatus(status);
return ret_status;