1syntax = "proto3"; 2 3package tensorflow.tensorrt; 4 5import "tensorflow/core/framework/tensor_shape.proto"; 6 7// Containing information for a serialized TensorRT engine. 8message TRTEngineInstance { 9 // The input shapes of the TRT engine. 10 repeated TensorShapeProto input_shapes = 1; 11 12 // The serialized TRT engine. 13 // 14 // TODO(laigd): consider using a more efficient in-memory representation 15 // instead of string which is the default here. 16 bytes serialized_engine = 2; 17 18 // TODO(laigd): consider adding calibration stats, precision_modes, etc. 19} 20