tensorflow.log_memory source code

1 // Generated by the protocol buffer compiler.  DO NOT EDIT!
2 // source: tensorflow/core/framework/log_memory.proto
3 
4 module tensorflow.log_memory;
5 
6 import google.protobuf;
7 import tensorflow.tensor_description;
8 
9 enum protocVersion = 3012004;
10 
11 class MemoryLogStep
12 {
13     @Proto(1) long stepId = protoDefaultValue!(long);
14     @Proto(2) string handle = protoDefaultValue!(string);
15 }
16 
17 class MemoryLogTensorAllocation
18 {
19     @Proto(1) long stepId = protoDefaultValue!(long);
20     @Proto(2) string kernelName = protoDefaultValue!(string);
21     @Proto(3) TensorDescription tensor = protoDefaultValue!(TensorDescription);
22 }
23 
24 class MemoryLogTensorDeallocation
25 {
26     @Proto(1) long allocationId = protoDefaultValue!(long);
27     @Proto(2) string allocatorName = protoDefaultValue!(string);
28 }
29 
30 class MemoryLogTensorOutput
31 {
32     @Proto(1) long stepId = protoDefaultValue!(long);
33     @Proto(2) string kernelName = protoDefaultValue!(string);
34     @Proto(3) int index = protoDefaultValue!(int);
35     @Proto(4) TensorDescription tensor = protoDefaultValue!(TensorDescription);
36 }
37 
38 class MemoryLogRawAllocation
39 {
40     @Proto(1) long stepId = protoDefaultValue!(long);
41     @Proto(2) string operation = protoDefaultValue!(string);
42     @Proto(3) long numBytes = protoDefaultValue!(long);
43     @Proto(4) ulong ptr = protoDefaultValue!(ulong);
44     @Proto(5) long allocationId = protoDefaultValue!(long);
45     @Proto(6) string allocatorName = protoDefaultValue!(string);
46 }
47 
48 class MemoryLogRawDeallocation
49 {
50     @Proto(1) long stepId = protoDefaultValue!(long);
51     @Proto(2) string operation = protoDefaultValue!(string);
52     @Proto(3) long allocationId = protoDefaultValue!(long);
53     @Proto(4) string allocatorName = protoDefaultValue!(string);
54     @Proto(5) bool deferred = protoDefaultValue!(bool);
55 }