Home
last modified time | relevance | path

Searched refs:cpu_allocator_collect_stats (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcpu_allocator_impl.cc32 static bool cpu_allocator_collect_stats = false; variable
34 void EnableCPUAllocatorStats() { cpu_allocator_collect_stats = true; } in EnableCPUAllocatorStats()
35 void DisableCPUAllocatorStats() { cpu_allocator_collect_stats = false; } in DisableCPUAllocatorStats()
36 bool CPUAllocatorStatsEnabled() { return cpu_allocator_collect_stats; } in CPUAllocatorStatsEnabled()
88 if (cpu_allocator_collect_stats) { in AllocateRaw()
113 if (cpu_allocator_collect_stats) { in DeallocateRaw()
148 if (!cpu_allocator_collect_stats) return absl::nullopt; in GetStats()
154 if (!cpu_allocator_collect_stats) return false; in ClearStats()