diff --git a/tensorflow/stream_executor/tpu/tpu_platform.cc b/tensorflow/stream_executor/tpu/tpu_platform.cc index 7a789d0cff0..d0340b1cc69 100644 --- a/tensorflow/stream_executor/tpu/tpu_platform.cc +++ b/tensorflow/stream_executor/tpu/tpu_platform.cc @@ -129,7 +129,7 @@ Status TpuPlatform::TpusPerHost(int* tpus) { Status TpuPlatform::TpuMemoryLimit(int64* memory_limit) { TF_Status* status = TF_NewStatus(); tpu::ConfigApiFn()->TpuConfigurationApi_TpuMemoryLimitFn( - reinterpret_cast(&memory_limit), status); + reinterpret_cast(memory_limit), status); auto ret_status = StatusFromTF_Status(status); TF_DeleteStatus(status); return ret_status;