Fix --config=asan leak reports.

PiperOrigin-RevId: 353077476
Change-Id: Iffaa02a4ea7caf182e918d8237eb1b5ee005394f
This commit is contained in:
A. Unique TensorFlower 2021-01-21 12:34:15 -08:00 committed by TensorFlower Gardener
parent 05ce78c1c5
commit 11e09d9352

View File

@ -4334,6 +4334,7 @@ EagerContextThreadLocalData* GetEagerContextThreadLocalData(
}
if (eager_context_thread_local_data_map == nullptr) {
absl::LeakCheckDisabler disabler;
eager_context_thread_local_data_map = new EagerContextThreadLocalDataMap();
}
auto& thread_local_data =