tensorflow.step_stats source code

1 // Generated by the protocol buffer compiler.  DO NOT EDIT!
2 // source: tensorflow/core/framework/step_stats.proto
3 
4 module tensorflow.step_stats;
5 
6 import google.protobuf;
7 import tensorflow.allocation_description;
8 import tensorflow.tensor_description;
9 
10 enum protocVersion = 3012004;
11 
12 class AllocationRecord
13 {
14     @Proto(1) long allocMicros = protoDefaultValue!(long);
15     @Proto(2) long allocBytes = protoDefaultValue!(long);
16 }
17 
18 class AllocatorMemoryUsed
19 {
20     @Proto(1) string allocatorName = protoDefaultValue!(string);
21     @Proto(2) long totalBytes = protoDefaultValue!(long);
22     @Proto(3) long peakBytes = protoDefaultValue!(long);
23     @Proto(4) long liveBytes = protoDefaultValue!(long);
24     @Proto(5) long allocatorBytesInUse = protoDefaultValue!(long);
25     @Proto(6) AllocationRecord[] allocationRecords = protoDefaultValue!(AllocationRecord[]);
26 }
27 
28 class NodeOutput
29 {
30     @Proto(1) int slot = protoDefaultValue!(int);
31     @Proto(3) TensorDescription tensorDescription = protoDefaultValue!(TensorDescription);
32 }
33 
34 class MemoryStats
35 {
36     @Proto(1) long tempMemorySize = protoDefaultValue!(long);
37     @Proto(2) long deviceTempMemorySize = protoDefaultValue!(long);
38     @Proto(3) long persistentMemorySize = protoDefaultValue!(long);
39     @Proto(4) long devicePersistentMemorySize = protoDefaultValue!(long);
40     @Proto(5, Wire.none, Yes.packed) long[] persistentTensorAllocIds = protoDefaultValue!(long[]);
41     @Proto(6) long[] devicePersistentTensorAllocIds = protoDefaultValue!(long[]);
42 }
43 
44 class NodeExecStats
45 {
46     @Proto(1) string nodeName = protoDefaultValue!(string);
47     @Proto(2) long allStartMicros = protoDefaultValue!(long);
48     @Proto(3) long opStartRelMicros = protoDefaultValue!(long);
49     @Proto(4) long opEndRelMicros = protoDefaultValue!(long);
50     @Proto(5) long allEndRelMicros = protoDefaultValue!(long);
51     @Proto(6) AllocatorMemoryUsed[] memory = protoDefaultValue!(AllocatorMemoryUsed[]);
52     @Proto(7) NodeOutput[] output = protoDefaultValue!(NodeOutput[]);
53     @Proto(8) string timelineLabel = protoDefaultValue!(string);
54     @Proto(9) long scheduledMicros = protoDefaultValue!(long);
55     @Proto(10) uint threadId = protoDefaultValue!(uint);
56     @Proto(11) AllocationDescription[] referencedTensor = protoDefaultValue!(AllocationDescription[]);
57     @Proto(12) MemoryStats memoryStats = protoDefaultValue!(MemoryStats);
58     @Proto(13) long allStartNanos = protoDefaultValue!(long);
59     @Proto(14) long opStartRelNanos = protoDefaultValue!(long);
60     @Proto(15) long opEndRelNanos = protoDefaultValue!(long);
61     @Proto(16) long allEndRelNanos = protoDefaultValue!(long);
62     @Proto(17) long scheduledNanos = protoDefaultValue!(long);
63 }
64 
65 class DeviceStepStats
66 {
67     @Proto(1) string device = protoDefaultValue!(string);
68     @Proto(2) NodeExecStats[] nodeStats = protoDefaultValue!(NodeExecStats[]);
69     @Proto(3) string[uint] threadNames = protoDefaultValue!(string[uint]);
70 }
71 
72 class StepStats
73 {
74     @Proto(1) DeviceStepStats[] devStats = protoDefaultValue!(DeviceStepStats[]);
75 }