v1

type File struct {

	// name of the file, with file extension
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// description of what the file contains
	Description string `protobuf:"bytes,2,opt,name=description,proto3" json:"description,omitempty"`
	// How to associate the arrays/tensors to the labels in the file
	LabelType LabelType `` /* 128-byte string literal not displayed */
	// contains filtered or unexported fields
}

type FlatTensor struct {

	// the shape of the provided tensor as a list of integer extents
	Shape []uint64 `protobuf:"fixed64,1,rep,packed,name=shape,proto3" json:"shape,omitempty"`
	// the flat data to be interpreted per the above shape information
	//
	// Types that are assignable to Tensor:
	//
	//	*FlatTensor_Int8Tensor
	//	*FlatTensor_Uint8Tensor
	//	*FlatTensor_Int16Tensor
	//	*FlatTensor_Uint16Tensor
	//	*FlatTensor_Int32Tensor
	//	*FlatTensor_Uint32Tensor
	//	*FlatTensor_Int64Tensor
	//	*FlatTensor_Uint64Tensor
	//	*FlatTensor_FloatTensor
	//	*FlatTensor_DoubleTensor
	Tensor isFlatTensor_Tensor `protobuf_oneof:"tensor"`
	// contains filtered or unexported fields
}

type FlatTensorDataDouble struct {
	Data []float64 `protobuf:"fixed64,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataFloat struct {
	Data []float32 `protobuf:"fixed32,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataInt16 struct {

	// packs two 16-bit numbers per entry - explicitly little-endian
	// so big-endian producers/consumers must compensate
	Data []uint32 `protobuf:"fixed32,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataInt32 struct {
	Data []int32 `protobuf:"fixed32,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataInt64 struct {
	Data []int64 `protobuf:"fixed64,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataInt8 struct {
	Data []byte `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataUInt16 struct {

	// packs two 16-bit numbers per entry - explicitly little-endian
	// so big-endian producers/consumers must compensate
	Data []uint32 `protobuf:"fixed32,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataUInt32 struct {
	Data []uint32 `protobuf:"fixed32,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataUInt64 struct {
	Data []uint64 `protobuf:"fixed64,1,rep,packed,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensorDataUInt8 struct {
	Data []byte `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type FlatTensor_DoubleTensor struct {
	DoubleTensor *FlatTensorDataDouble `protobuf:"bytes,11,opt,name=double_tensor,json=doubleTensor,proto3,oneof"`
}

type FlatTensor_FloatTensor struct {
	FloatTensor *FlatTensorDataFloat `protobuf:"bytes,10,opt,name=float_tensor,json=floatTensor,proto3,oneof"`
}

type FlatTensor_Int16Tensor struct {
	Int16Tensor *FlatTensorDataInt16 `protobuf:"bytes,4,opt,name=int16_tensor,json=int16Tensor,proto3,oneof"`
}

type FlatTensor_Int32Tensor struct {
	Int32Tensor *FlatTensorDataInt32 `protobuf:"bytes,6,opt,name=int32_tensor,json=int32Tensor,proto3,oneof"`
}

type FlatTensor_Int64Tensor struct {
	Int64Tensor *FlatTensorDataInt64 `protobuf:"bytes,8,opt,name=int64_tensor,json=int64Tensor,proto3,oneof"`
}

type FlatTensor_Int8Tensor struct {
	Int8Tensor *FlatTensorDataInt8 `protobuf:"bytes,2,opt,name=int8_tensor,json=int8Tensor,proto3,oneof"`
}

type FlatTensor_Uint16Tensor struct {
	Uint16Tensor *FlatTensorDataUInt16 `protobuf:"bytes,5,opt,name=uint16_tensor,json=uint16Tensor,proto3,oneof"`
}

type FlatTensor_Uint32Tensor struct {
	Uint32Tensor *FlatTensorDataUInt32 `protobuf:"bytes,7,opt,name=uint32_tensor,json=uint32Tensor,proto3,oneof"`
}

type FlatTensor_Uint64Tensor struct {
	Uint64Tensor *FlatTensorDataUInt64 `protobuf:"bytes,9,opt,name=uint64_tensor,json=uint64Tensor,proto3,oneof"`
}

type FlatTensor_Uint8Tensor struct {
	Uint8Tensor *FlatTensorDataUInt8 `protobuf:"bytes,3,opt,name=uint8_tensor,json=uint8Tensor,proto3,oneof"`
}

type FlatTensors struct {

	// A name-indexed collection of flat tensor objects
	Tensors map[string]*FlatTensor `` /* 155-byte string literal not displayed */
	// contains filtered or unexported fields
}

type InferRequest struct {

	// name of the model service
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// the input data is provided as set of named flat tensors
	InputTensors *FlatTensors `protobuf:"bytes,3,opt,name=input_tensors,json=inputTensors,proto3" json:"input_tensors,omitempty"`
	// Additional arguments to the method
	Extra *structpb.Struct `protobuf:"bytes,99,opt,name=extra,proto3" json:"extra,omitempty"`
	// contains filtered or unexported fields
}

type InferResponse struct {

	// the output data is provided as a set of named flat tensors
	OutputTensors *FlatTensors `protobuf:"bytes,3,opt,name=output_tensors,json=outputTensors,proto3" json:"output_tensors,omitempty"`
	// contains filtered or unexported fields
}

type LabelType int32

type MLModelServiceClient interface {
	// Infer takes an already ordered input tensor as a map, makes an inference on the model, and returns an output data map.
	Infer(ctx context.Context, in *InferRequest, opts ...grpc.CallOption) (*InferResponse, error)
	// Metadata returns the metadata associated with the ML model.
	Metadata(ctx context.Context, in *MetadataRequest, opts ...grpc.CallOption) (*MetadataResponse, error)
}

type MLModelServiceServer interface {
	// Infer takes an already ordered input tensor as a map, makes an inference on the model, and returns an output data map.
	Infer(context.Context, *InferRequest) (*InferResponse, error)
	// Metadata returns the metadata associated with the ML model.
	Metadata(context.Context, *MetadataRequest) (*MetadataResponse, error)
	// contains filtered or unexported methods
}

type Metadata struct {

	// name of the model
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// type of model e.g. object_detector, text_classifier
	Type string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"`
	// description of the model
	Description string `protobuf:"bytes,3,opt,name=description,proto3" json:"description,omitempty"`
	// the necessary input arrays/tensors for an inference, order matters
	InputInfo []*TensorInfo `protobuf:"bytes,4,rep,name=input_info,json=inputInfo,proto3" json:"input_info,omitempty"`
	// the output arrays/tensors of the model, order matters
	OutputInfo []*TensorInfo `protobuf:"bytes,5,rep,name=output_info,json=outputInfo,proto3" json:"output_info,omitempty"`
	// contains filtered or unexported fields
}

type MetadataRequest struct {

	// name of the model service
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Additional arguments to the method
	Extra *structpb.Struct `protobuf:"bytes,99,opt,name=extra,proto3" json:"extra,omitempty"`
	// contains filtered or unexported fields
}

type MetadataResponse struct {

	// this is the metadata associated with the ML model
	Metadata *Metadata `protobuf:"bytes,1,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// contains filtered or unexported fields
}

type TensorInfo struct {

	// name of the data in the array/tensor
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// description of the data in the array/tensor
	Description string `protobuf:"bytes,2,opt,name=description,proto3" json:"description,omitempty"`
	// data type of the array/tensor, e.g. float32, float64, uint8
	DataType string `protobuf:"bytes,3,opt,name=data_type,json=dataType,proto3" json:"data_type,omitempty"`
	// shape of the array/tensor (-1 for unknown)
	Shape []int32 `protobuf:"varint,4,rep,packed,name=shape,proto3" json:"shape,omitempty"`
	// files associated with the array/tensor, like for category labels
	AssociatedFiles []*File `protobuf:"bytes,5,rep,name=associated_files,json=associatedFiles,proto3" json:"associated_files,omitempty"`
	// anything else you want to say
	Extra *structpb.Struct `protobuf:"bytes,99,opt,name=extra,proto3" json:"extra,omitempty"`
	// contains filtered or unexported fields
}

type UnimplementedMLModelServiceServer struct {
}

type UnsafeMLModelServiceServer interface {
	// contains filtered or unexported methods
}