torch_predict_protos

package
v2.1.9 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: May 13, 2024 License: Apache-2.0 Imports: 3 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var ArrayDataType_name = map[int32]string{
	0:  "DT_INVALID",
	1:  "DT_FLOAT",
	2:  "DT_DOUBLE",
	3:  "DT_INT32",
	4:  "DT_UINT8",
	5:  "DT_INT16",
	6:  "DT_INT8",
	7:  "DT_STRING",
	8:  "DT_COMPLEX64",
	9:  "DT_INT64",
	10: "DT_BOOL",
	11: "DT_QINT8",
	12: "DT_QUINT8",
	13: "DT_QINT32",
	14: "DT_BFLOAT16",
	15: "DT_QINT16",
	16: "DT_QUINT16",
	17: "DT_UINT16",
	18: "DT_COMPLEX128",
	19: "DT_HALF",
	20: "DT_RESOURCE",
	21: "DT_VARIANT",
}
View Source 
var ArrayDataType_value = map[string]int32{
	"DT_INVALID":    0,
	"DT_FLOAT":      1,
	"DT_DOUBLE":     2,
	"DT_INT32":      3,
	"DT_UINT8":      4,
	"DT_INT16":      5,
	"DT_INT8":       6,
	"DT_STRING":     7,
	"DT_COMPLEX64":  8,
	"DT_INT64":      9,
	"DT_BOOL":       10,
	"DT_QINT8":      11,
	"DT_QUINT8":     12,
	"DT_QINT32":     13,
	"DT_BFLOAT16":   14,
	"DT_QINT16":     15,
	"DT_QUINT16":    16,
	"DT_UINT16":     17,
	"DT_COMPLEX128": 18,
	"DT_HALF":       19,
	"DT_RESOURCE":   20,
	"DT_VARIANT":    21,
}

Functions

This section is empty.

Types

type ArrayDataType 

type ArrayDataType int32
const (
	// Not a legal value for DataType. Used to indicate a DataType field
	// has not been set.
	ArrayDataType_DT_INVALID ArrayDataType = 0
	// Data types that all computation devices are expected to be
	// capable to support.
	ArrayDataType_DT_FLOAT      ArrayDataType = 1
	ArrayDataType_DT_DOUBLE     ArrayDataType = 2
	ArrayDataType_DT_INT32      ArrayDataType = 3
	ArrayDataType_DT_UINT8      ArrayDataType = 4
	ArrayDataType_DT_INT16      ArrayDataType = 5
	ArrayDataType_DT_INT8       ArrayDataType = 6
	ArrayDataType_DT_STRING     ArrayDataType = 7
	ArrayDataType_DT_COMPLEX64  ArrayDataType = 8
	ArrayDataType_DT_INT64      ArrayDataType = 9
	ArrayDataType_DT_BOOL       ArrayDataType = 10
	ArrayDataType_DT_QINT8      ArrayDataType = 11
	ArrayDataType_DT_QUINT8     ArrayDataType = 12
	ArrayDataType_DT_QINT32     ArrayDataType = 13
	ArrayDataType_DT_BFLOAT16   ArrayDataType = 14
	ArrayDataType_DT_QINT16     ArrayDataType = 15
	ArrayDataType_DT_QUINT16    ArrayDataType = 16
	ArrayDataType_DT_UINT16     ArrayDataType = 17
	ArrayDataType_DT_COMPLEX128 ArrayDataType = 18
	ArrayDataType_DT_HALF       ArrayDataType = 19
	ArrayDataType_DT_RESOURCE   ArrayDataType = 20
	ArrayDataType_DT_VARIANT    ArrayDataType = 21
)

func (ArrayDataType) EnumDescriptor 

func (ArrayDataType) EnumDescriptor() ([]byte, []int)

func (ArrayDataType) String 

func (x ArrayDataType) String() string

type ArrayProto 

type ArrayProto struct {
	// Data Type.
	Dtype ArrayDataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=pytorch.eas.ArrayDataType" json:"dtype,omitempty"`
	// Shape of the array.
	ArrayShape *ArrayShape `protobuf:"bytes,2,opt,name=array_shape,json=arrayShape,proto3" json:"array_shape,omitempty"`
	// DT_FLOAT.
	FloatVal []float32 `protobuf:"fixed32,3,rep,packed,name=float_val,json=floatVal,proto3" json:"float_val,omitempty"`
	// DT_DOUBLE.
	DoubleVal []float64 `protobuf:"fixed64,4,rep,packed,name=double_val,json=doubleVal,proto3" json:"double_val,omitempty"`
	// DT_INT32, DT_INT16, DT_INT8, DT_UINT8.
	IntVal []int32 `protobuf:"varint,5,rep,packed,name=int_val,json=intVal,proto3" json:"int_val,omitempty"`
	// DT_STRING.
	StringVal [][]byte `protobuf:"bytes,6,rep,name=string_val,json=stringVal,proto3" json:"string_val,omitempty"`
	// DT_INT64.
	Int64Val             []int64  `protobuf:"varint,7,rep,packed,name=int64_val,json=int64Val,proto3" json:"int64_val,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Protocol buffer representing an array

func (*ArrayProto) Descriptor 

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

func (*ArrayProto) GetArrayShape 

func (m *ArrayProto) GetArrayShape() *ArrayShape

func (*ArrayProto) GetDoubleVal 

func (m *ArrayProto) GetDoubleVal() []float64

func (*ArrayProto) GetDtype 

func (m *ArrayProto) GetDtype() ArrayDataType

func (*ArrayProto) GetFloatVal 

func (m *ArrayProto) GetFloatVal() []float32

func (*ArrayProto) GetInt64Val 

func (m *ArrayProto) GetInt64Val() []int64

func (*ArrayProto) GetIntVal 

func (m *ArrayProto) GetIntVal() []int32

func (*ArrayProto) GetStringVal 

func (m *ArrayProto) GetStringVal() [][]byte

func (*ArrayProto) ProtoMessage 

func (*ArrayProto) ProtoMessage()

func (*ArrayProto) Reset 

func (m *ArrayProto) Reset()

func (*ArrayProto) String 

func (m *ArrayProto) String() string

func (*ArrayProto) XXX_DiscardUnknown 

func (m *ArrayProto) XXX_DiscardUnknown()

func (*ArrayProto) XXX_Marshal 

func (m *ArrayProto) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*ArrayProto) XXX_Merge 

func (m *ArrayProto) XXX_Merge(src proto.Message)

func (*ArrayProto) XXX_Size 

func (m *ArrayProto) XXX_Size() int

func (*ArrayProto) XXX_Unmarshal 

func (m *ArrayProto) XXX_Unmarshal(b []byte) error

type ArrayShape 

type ArrayShape struct {
	Dim                  []int64  `protobuf:"varint,1,rep,packed,name=dim,proto3" json:"dim,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

Dimensions of an array

func (*ArrayShape) Descriptor 

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

func (*ArrayShape) GetDim 

func (m *ArrayShape) GetDim() []int64

func (*ArrayShape) ProtoMessage 

func (*ArrayShape) ProtoMessage()

func (*ArrayShape) Reset 

func (m *ArrayShape) Reset()

func (*ArrayShape) String 

func (m *ArrayShape) String() string

func (*ArrayShape) XXX_DiscardUnknown 

func (m *ArrayShape) XXX_DiscardUnknown()

func (*ArrayShape) XXX_Marshal 

func (m *ArrayShape) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*ArrayShape) XXX_Merge 

func (m *ArrayShape) XXX_Merge(src proto.Message)

func (*ArrayShape) XXX_Size 

func (m *ArrayShape) XXX_Size() int

func (*ArrayShape) XXX_Unmarshal 

func (m *ArrayShape) XXX_Unmarshal(b []byte) error

type PredictRequest 

type PredictRequest struct {
	// Input tensors.
	Inputs []*ArrayProto `protobuf:"bytes,1,rep,name=inputs,proto3" json:"inputs,omitempty"`
	// Output filter.
	OutputFilter         []int32  `protobuf:"varint,2,rep,packed,name=output_filter,json=outputFilter,proto3" json:"output_filter,omitempty"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

func (*PredictRequest) Descriptor 

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

func (*PredictRequest) GetInputs 

func (m *PredictRequest) GetInputs() []*ArrayProto

func (*PredictRequest) GetOutputFilter 

func (m *PredictRequest) GetOutputFilter() []int32

func (*PredictRequest) ProtoMessage 

func (*PredictRequest) ProtoMessage()

func (*PredictRequest) Reset 

func (m *PredictRequest) Reset()

func (*PredictRequest) String 

func (m *PredictRequest) String() string

func (*PredictRequest) XXX_DiscardUnknown 

func (m *PredictRequest) XXX_DiscardUnknown()

func (*PredictRequest) XXX_Marshal 

func (m *PredictRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*PredictRequest) XXX_Merge 

func (m *PredictRequest) XXX_Merge(src proto.Message)

func (*PredictRequest) XXX_Size 

func (m *PredictRequest) XXX_Size() int

func (*PredictRequest) XXX_Unmarshal 

func (m *PredictRequest) XXX_Unmarshal(b []byte) error

type PredictResponse 

type PredictResponse struct {
	// Output tensors.
	Outputs              []*ArrayProto `protobuf:"bytes,1,rep,name=outputs,proto3" json:"outputs,omitempty"`
	XXX_NoUnkeyedLiteral struct{}      `json:"-"`
	XXX_unrecognized     []byte        `json:"-"`
	XXX_sizecache        int32         `json:"-"`
}

Response for PredictRequest on successful run.

func (*PredictResponse) Descriptor 

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

func (*PredictResponse) GetOutputs 

func (m *PredictResponse) GetOutputs() []*ArrayProto

func (*PredictResponse) ProtoMessage 

func (*PredictResponse) ProtoMessage()

func (*PredictResponse) Reset 

func (m *PredictResponse) Reset()

func (*PredictResponse) String 

func (m *PredictResponse) String() string

func (*PredictResponse) XXX_DiscardUnknown 

func (m *PredictResponse) XXX_DiscardUnknown()

func (*PredictResponse) XXX_Marshal 

func (m *PredictResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*PredictResponse) XXX_Merge 

func (m *PredictResponse) XXX_Merge(src proto.Message)

func (*PredictResponse) XXX_Size 

func (m *PredictResponse) XXX_Size() int

func (*PredictResponse) XXX_Unmarshal 

func (m *PredictResponse) XXX_Unmarshal(b []byte) error

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.