1syntax = "proto3"; 2 3package tensorflow; 4 5option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto"; 6 7// Some of the data from AllocatorStats 8message MemAllocatorStats { 9 int64 num_allocs = 1; 10 int64 bytes_in_use = 2; 11 int64 peak_bytes_in_use = 3; 12 int64 largest_alloc_size = 4; 13 float fragmentation_metric = 5; 14} 15 16message MemChunk { 17 uint64 address = 1; 18 int64 size = 2; 19 int64 requested_size = 3; 20 int32 bin = 4; 21 string op_name = 5; 22 uint64 freed_at_count = 6; 23 uint64 action_count = 7; 24 bool in_use = 8; 25 uint64 step_id = 9; 26} 27 28message BinSummary { 29 int32 bin = 1; 30 int64 total_bytes_in_use = 2; 31 int64 total_bytes_in_bin = 3; 32 int64 total_chunks_in_use = 4; 33 int64 total_chunks_in_bin = 5; 34} 35 36message SnapShot { 37 uint64 action_count = 1; 38 int64 size = 2; 39} 40 41message MemoryDump { 42 string allocator_name = 1; 43 repeated BinSummary bin_summary = 2; 44 repeated MemChunk chunk = 3; 45 repeated SnapShot snap_shot = 4; 46 MemAllocatorStats stats = 5; 47} 48