xref: /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/trt_engine_instance.proto (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
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