1syntax = "proto2"; 2 3package tensorflow; 4 5import "tensorflow/core/framework/node_def.proto"; 6import "tensorflow/core/framework/tensor.proto"; 7import "tensorflow/core/framework/tensor_shape.proto"; 8import "tensorflow/core/framework/types.proto"; 9 10message TfCallbackData { 11 message BufferDescription { 12 optional TensorShapeProto shape = 1; 13 optional DataType type = 2; 14 } 15 16 message InputBufferDescription { 17 optional BufferDescription buffer_description = 1; 18 19 // The input value might be already fixed at the compilation time. 20 // This value may or may not be present. 21 optional TensorProto value = 2; 22 } 23 24 message OutputBufferDescription { 25 optional BufferDescription buffer_description = 1; 26 // Whether the buffer stores dynamically padded data: in that case, actual 27 // concrete dimensions need to be stored after the buffer. 28 optional bool is_dynamically_padded = 2; 29 } 30 31 optional tensorflow.NodeDef op = 1; 32 repeated InputBufferDescription inputs = 2; 33 repeated OutputBufferDescription outputs = 3; 34} 35