diff options
Diffstat (limited to 'tensorflow/core/framework/allocator.h')
-rw-r--r-- | tensorflow/core/framework/allocator.h | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tensorflow/core/framework/allocator.h b/tensorflow/core/framework/allocator.h index 2bb4d32d57..774b1fe137 100644 --- a/tensorflow/core/framework/allocator.h +++ b/tensorflow/core/framework/allocator.h @@ -376,16 +376,18 @@ struct AllocatorAttributes { int32 scope_id = 0; }; -// Returns a trivial implementation of Allocator which uses the system -// default malloc. The returned allocator is a process singleton. +// Returns a trivial implementation of Allocator, which is a process singleton. +// Access through this function is only intended for use in tests and auxiliary +// processing. Performance sensitive uses should always obtain allocators from +// ProcessState. Allocator* cpu_allocator(); -// If 'enable' is true, the process-wide cpu allocator collects +// If 'enable' is true, the default CPU allocator implementation will collect // AllocatorStats. By default, it's disabled. void EnableCPUAllocatorStats(bool enable); -// If 'enable' is true, the process-wide cpu allocator collects full -// statistics. By default, it's disabled. +// If 'enable' is true, the default CPU allocator implementation will collect +// full statistics. By default, it's disabled. void EnableCPUAllocatorFullStats(bool enable); // Abstract interface of an object that does the underlying suballoc/free of |