utils

Documentation

Index

• Variables
• type TRTEngineInstance
  • func (*TRTEngineInstance) Descriptor() ([]byte, []int)
  • func (x *TRTEngineInstance) GetInputShapes() []*framework.TensorShapeProto
  • func (x *TRTEngineInstance) GetSerializedEngine() []byte
  • func (*TRTEngineInstance) ProtoMessage()
  • func (x *TRTEngineInstance) ProtoReflect() protoreflect.Message
  • func (x *TRTEngineInstance) Reset()
  • func (x *TRTEngineInstance) String() string

Variables

var File_tensorflow_compiler_tf2tensorrt_utils_trt_engine_instance_proto protoreflect.FileDescriptor

Types

type TRTEngineInstance

type TRTEngineInstance struct {

	// The input shapes of the TRT engine.
	InputShapes []*framework.TensorShapeProto `protobuf:"bytes,1,rep,name=input_shapes,json=inputShapes,proto3" json:"input_shapes,omitempty"`
	// The serialized TRT engine.
	//
	// TODO(laigd): consider using a more efficient in-memory representation
	// instead of string which is the default here.
	SerializedEngine []byte `protobuf:"bytes,2,opt,name=serialized_engine,json=serializedEngine,proto3" json:"serialized_engine,omitempty"`
	// contains filtered or unexported fields
}

Containing information for a serialized TensorRT engine.

func (*TRTEngineInstance) Descriptor

func (*TRTEngineInstance) Descriptor() ([]byte, []int)

Deprecated: Use TRTEngineInstance.ProtoReflect.Descriptor instead.

func (*TRTEngineInstance) GetInputShapes

func (x *TRTEngineInstance) GetInputShapes() []*framework.TensorShapeProto

func (*TRTEngineInstance) GetSerializedEngine

func (x *TRTEngineInstance) GetSerializedEngine() []byte

func (*TRTEngineInstance) ProtoMessage

func (*TRTEngineInstance) ProtoMessage()

func (*TRTEngineInstance) ProtoReflect

func (x *TRTEngineInstance) ProtoReflect() protoreflect.Message

func (*TRTEngineInstance) Reset

func (x *TRTEngineInstance) Reset()

func (*TRTEngineInstance) String

func (x *TRTEngineInstance) String() string