pb

package
v1.0.1 Latest Latest
Warning

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

Go to latest
Published: Jun 30, 2026 License: MIT Imports: 14 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source
var AttributeProto_AttributeType_name = map[int32]string{
	0:  "UNDEFINED",
	1:  "FLOAT",
	2:  "INT",
	3:  "STRING",
	4:  "TENSOR",
	5:  "GRAPH",
	11: "SPARSE_TENSOR",
	13: "TYPE_PROTO",
	6:  "FLOATS",
	7:  "INTS",
	8:  "STRINGS",
	9:  "TENSORS",
	10: "GRAPHS",
	12: "SPARSE_TENSORS",
	14: "TYPE_PROTOS",
}
View Source
var AttributeProto_AttributeType_value = map[string]int32{
	"UNDEFINED":      0,
	"FLOAT":          1,
	"INT":            2,
	"STRING":         3,
	"TENSOR":         4,
	"GRAPH":          5,
	"SPARSE_TENSOR":  11,
	"TYPE_PROTO":     13,
	"FLOATS":         6,
	"INTS":           7,
	"STRINGS":        8,
	"TENSORS":        9,
	"GRAPHS":         10,
	"SPARSE_TENSORS": 12,
	"TYPE_PROTOS":    14,
}
View Source
var File_onnx_proto3 *descriptor.FileDescriptor
View Source
var OperatorStatus_name = map[int32]string{
	0: "EXPERIMENTAL",
	1: "STABLE",
}
View Source
var OperatorStatus_value = map[string]int32{
	"EXPERIMENTAL": 0,
	"STABLE":       1,
}
View Source
var TensorProto_DataLocation_name = map[int32]string{
	0: "DEFAULT",
	1: "EXTERNAL",
}
View Source
var TensorProto_DataLocation_value = map[string]int32{
	"DEFAULT":  0,
	"EXTERNAL": 1,
}
View Source
var TensorProto_DataType_name = map[int32]string{
	0:  "UNDEFINED",
	1:  "FLOAT",
	2:  "UINT8",
	3:  "INT8",
	4:  "UINT16",
	5:  "INT16",
	6:  "INT32",
	7:  "INT64",
	8:  "STRING",
	9:  "BOOL",
	10: "FLOAT16",
	11: "DOUBLE",
	12: "UINT32",
	13: "UINT64",
	14: "COMPLEX64",
	15: "COMPLEX128",
	16: "BFLOAT16",
	17: "FLOAT8E4M3FN",
	18: "FLOAT8E4M3FNUZ",
	19: "FLOAT8E5M2",
	20: "FLOAT8E5M2FNUZ",
	21: "UINT4",
	22: "INT4",
	23: "FLOAT4E2M1",
	24: "FLOAT8E8M0",
	25: "UINT2",
	26: "INT2",
}
View Source
var TensorProto_DataType_value = map[string]int32{
	"UNDEFINED":      0,
	"FLOAT":          1,
	"UINT8":          2,
	"INT8":           3,
	"UINT16":         4,
	"INT16":          5,
	"INT32":          6,
	"INT64":          7,
	"STRING":         8,
	"BOOL":           9,
	"FLOAT16":        10,
	"DOUBLE":         11,
	"UINT32":         12,
	"UINT64":         13,
	"COMPLEX64":      14,
	"COMPLEX128":     15,
	"BFLOAT16":       16,
	"FLOAT8E4M3FN":   17,
	"FLOAT8E4M3FNUZ": 18,
	"FLOAT8E5M2":     19,
	"FLOAT8E5M2FNUZ": 20,
	"UINT4":          21,
	"INT4":           22,
	"FLOAT4E2M1":     23,
	"FLOAT8E8M0":     24,
	"UINT2":          25,
	"INT2":           26,
}
View Source
var Version_name = map[int32]string{
	0:  "_START_VERSION",
	1:  "IR_VERSION_2017_10_10",
	2:  "IR_VERSION_2017_10_30",
	3:  "IR_VERSION_2017_11_3",
	4:  "IR_VERSION_2019_1_22",
	5:  "IR_VERSION_2019_3_18",
	6:  "IR_VERSION_2019_9_19",
	7:  "IR_VERSION_2020_5_8",
	8:  "IR_VERSION_2021_7_30",
	9:  "IR_VERSION_2023_5_5",
	10: "IR_VERSION_2024_3_25",
	11: "IR_VERSION_2025_05_12",
	12: "IR_VERSION_2025_08_26",
	13: "IR_VERSION",
}
View Source
var Version_value = map[string]int32{
	"_START_VERSION":        0,
	"IR_VERSION_2017_10_10": 1,
	"IR_VERSION_2017_10_30": 2,
	"IR_VERSION_2017_11_3":  3,
	"IR_VERSION_2019_1_22":  4,
	"IR_VERSION_2019_3_18":  5,
	"IR_VERSION_2019_9_19":  6,
	"IR_VERSION_2020_5_8":   7,
	"IR_VERSION_2021_7_30":  8,
	"IR_VERSION_2023_5_5":   9,
	"IR_VERSION_2024_3_25":  10,
	"IR_VERSION_2025_05_12": 11,
	"IR_VERSION_2025_08_26": 12,
	"IR_VERSION":            13,
}

Functions

This section is empty.

Types

type AttributeProto

type AttributeProto struct {
	Name          string
	RefAttrName   string
	DocString     string
	Type          int32
	F             float32
	I             int64
	S             []byte
	T             *TensorProto
	G             *GraphProto
	SparseTensor  *SparseTensorProto
	Tp            *TypeProto
	Floats        []float32
	Ints          []int64
	Strings       [][]byte
	Tensors       []*TensorProto
	Graphs        []*GraphProto
	SparseTensors []*SparseTensorProto
	TypeProtos    []*TypeProto
	// contains filtered or unexported fields
}

func AttributeProtoFromPool

func AttributeProtoFromPool() *AttributeProto

func NewAttributeProto

func NewAttributeProto(opts ...AttributeProtoOption) *AttributeProto

func (*AttributeProto) Clone

func (x *AttributeProto) Clone() proto.Message

func (*AttributeProto) Equal

func (x *AttributeProto) Equal(other *AttributeProto) bool

func (*AttributeProto) GetDocString

func (x *AttributeProto) GetDocString() string

func (*AttributeProto) GetF

func (x *AttributeProto) GetF() float32

func (*AttributeProto) GetFloats

func (x *AttributeProto) GetFloats() []float32

func (*AttributeProto) GetG

func (x *AttributeProto) GetG() *GraphProto

func (*AttributeProto) GetGraphs

func (x *AttributeProto) GetGraphs() []*GraphProto

func (*AttributeProto) GetI

func (x *AttributeProto) GetI() int64

func (*AttributeProto) GetInts

func (x *AttributeProto) GetInts() []int64

func (*AttributeProto) GetName

func (x *AttributeProto) GetName() string

func (*AttributeProto) GetRefAttrName

func (x *AttributeProto) GetRefAttrName() string

func (*AttributeProto) GetS

func (x *AttributeProto) GetS() []byte

func (*AttributeProto) GetSparseTensor

func (x *AttributeProto) GetSparseTensor() *SparseTensorProto

func (*AttributeProto) GetSparseTensors

func (x *AttributeProto) GetSparseTensors() []*SparseTensorProto

func (*AttributeProto) GetStrings

func (x *AttributeProto) GetStrings() [][]byte

func (*AttributeProto) GetT

func (x *AttributeProto) GetT() *TensorProto

func (*AttributeProto) GetTensors

func (x *AttributeProto) GetTensors() []*TensorProto

func (*AttributeProto) GetTp

func (x *AttributeProto) GetTp() *TypeProto

func (*AttributeProto) GetType

func (x *AttributeProto) GetType() int32

func (*AttributeProto) GetTypeProtos

func (x *AttributeProto) GetTypeProtos() []*TypeProto

func (*AttributeProto) Marshal

func (x *AttributeProto) Marshal() ([]byte, error)

func (*AttributeProto) MarshalAppend

func (x *AttributeProto) MarshalAppend(b []byte) ([]byte, error)

func (*AttributeProto) MarshalBinary

func (x *AttributeProto) MarshalBinary() ([]byte, error)

func (*AttributeProto) MarshalJSON

func (x *AttributeProto) MarshalJSON() ([]byte, error)

func (*AttributeProto) Merge

func (x *AttributeProto) Merge(src proto.Message) error

func (*AttributeProto) ProtoReflect

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

func (*AttributeProto) Reset

func (x *AttributeProto) Reset()

func (*AttributeProto) ResetVT

func (x *AttributeProto) ResetVT()

func (*AttributeProto) ReturnToPool

func (x *AttributeProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*AttributeProto) ShallowClone

func (x *AttributeProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*AttributeProto) Size

func (x *AttributeProto) Size() int

func (*AttributeProto) Unmarshal

func (x *AttributeProto) Unmarshal(b []byte) error

func (*AttributeProto) UnmarshalBinary

func (x *AttributeProto) UnmarshalBinary(data []byte) error

func (*AttributeProto) UnmarshalJSON

func (x *AttributeProto) UnmarshalJSON(data []byte) error

func (*AttributeProto) UnmarshalUnsafe

func (x *AttributeProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*AttributeProto) Validate

func (x *AttributeProto) Validate() error

type AttributeProtoBuilder

type AttributeProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewAttributeProtoBuilder

func NewAttributeProtoBuilder() *AttributeProtoBuilder

func (*AttributeProtoBuilder) Build

func (*AttributeProtoBuilder) SetDocString

func (*AttributeProtoBuilder) SetF

func (*AttributeProtoBuilder) SetFloats

func (*AttributeProtoBuilder) SetG

func (*AttributeProtoBuilder) SetGraphs

func (*AttributeProtoBuilder) SetI

func (*AttributeProtoBuilder) SetInts

func (*AttributeProtoBuilder) SetName

func (*AttributeProtoBuilder) SetRefAttrName

func (b *AttributeProtoBuilder) SetRefAttrName(v string) *AttributeProtoBuilder

func (*AttributeProtoBuilder) SetS

func (*AttributeProtoBuilder) SetSparseTensor

func (*AttributeProtoBuilder) SetSparseTensors

func (*AttributeProtoBuilder) SetStrings

func (b *AttributeProtoBuilder) SetStrings(v [][]byte) *AttributeProtoBuilder

func (*AttributeProtoBuilder) SetT

func (*AttributeProtoBuilder) SetTensors

func (*AttributeProtoBuilder) SetTp

func (*AttributeProtoBuilder) SetType

func (*AttributeProtoBuilder) SetTypeProtos

func (b *AttributeProtoBuilder) SetTypeProtos(v []*TypeProto) *AttributeProtoBuilder

type AttributeProtoOption

type AttributeProtoOption func(*AttributeProto)

func AttributeProtoWithDocString

func AttributeProtoWithDocString(v string) AttributeProtoOption

func AttributeProtoWithF

func AttributeProtoWithF(v float32) AttributeProtoOption

func AttributeProtoWithFloats

func AttributeProtoWithFloats(v []float32) AttributeProtoOption

func AttributeProtoWithG

func AttributeProtoWithG(v *GraphProto) AttributeProtoOption

func AttributeProtoWithGraphs

func AttributeProtoWithGraphs(v []*GraphProto) AttributeProtoOption

func AttributeProtoWithI

func AttributeProtoWithI(v int64) AttributeProtoOption

func AttributeProtoWithInts

func AttributeProtoWithInts(v []int64) AttributeProtoOption

func AttributeProtoWithName

func AttributeProtoWithName(v string) AttributeProtoOption

func AttributeProtoWithRefAttrName

func AttributeProtoWithRefAttrName(v string) AttributeProtoOption

func AttributeProtoWithS

func AttributeProtoWithS(v []byte) AttributeProtoOption

func AttributeProtoWithSparseTensor

func AttributeProtoWithSparseTensor(v *SparseTensorProto) AttributeProtoOption

func AttributeProtoWithSparseTensors

func AttributeProtoWithSparseTensors(v []*SparseTensorProto) AttributeProtoOption

func AttributeProtoWithStrings

func AttributeProtoWithStrings(v [][]byte) AttributeProtoOption

func AttributeProtoWithT

func AttributeProtoWithT(v *TensorProto) AttributeProtoOption

func AttributeProtoWithTensors

func AttributeProtoWithTensors(v []*TensorProto) AttributeProtoOption

func AttributeProtoWithTp

func AttributeProtoWithTp(v *TypeProto) AttributeProtoOption

func AttributeProtoWithType

func AttributeProtoWithType(v int32) AttributeProtoOption

func AttributeProtoWithTypeProtos

func AttributeProtoWithTypeProtos(v []*TypeProto) AttributeProtoOption

type AttributeProto_AttributeType

type AttributeProto_AttributeType int32
const (
	AttributeProto_AttributeType_UNDEFINED      AttributeProto_AttributeType = 0
	AttributeProto_AttributeType_FLOAT          AttributeProto_AttributeType = 1
	AttributeProto_AttributeType_INT            AttributeProto_AttributeType = 2
	AttributeProto_AttributeType_STRING         AttributeProto_AttributeType = 3
	AttributeProto_AttributeType_TENSOR         AttributeProto_AttributeType = 4
	AttributeProto_AttributeType_GRAPH          AttributeProto_AttributeType = 5
	AttributeProto_AttributeType_SPARSE_TENSOR  AttributeProto_AttributeType = 11
	AttributeProto_AttributeType_TYPE_PROTO     AttributeProto_AttributeType = 13
	AttributeProto_AttributeType_FLOATS         AttributeProto_AttributeType = 6
	AttributeProto_AttributeType_INTS           AttributeProto_AttributeType = 7
	AttributeProto_AttributeType_STRINGS        AttributeProto_AttributeType = 8
	AttributeProto_AttributeType_TENSORS        AttributeProto_AttributeType = 9
	AttributeProto_AttributeType_GRAPHS         AttributeProto_AttributeType = 10
	AttributeProto_AttributeType_SPARSE_TENSORS AttributeProto_AttributeType = 12
	AttributeProto_AttributeType_TYPE_PROTOS    AttributeProto_AttributeType = 14
)

func ParseAttributeProto_AttributeType

func ParseAttributeProto_AttributeType(name string) (AttributeProto_AttributeType, bool)

func (AttributeProto_AttributeType) Number

func (AttributeProto_AttributeType) String

type DeviceConfigurationProto

type DeviceConfigurationProto struct {
	Name       string
	NumDevices int32
	Device     []string
	// contains filtered or unexported fields
}

func DeviceConfigurationProtoFromPool

func DeviceConfigurationProtoFromPool() *DeviceConfigurationProto

func (*DeviceConfigurationProto) Clone

func (*DeviceConfigurationProto) Equal

func (*DeviceConfigurationProto) GetDevice

func (x *DeviceConfigurationProto) GetDevice() []string

func (*DeviceConfigurationProto) GetName

func (x *DeviceConfigurationProto) GetName() string

func (*DeviceConfigurationProto) GetNumDevices

func (x *DeviceConfigurationProto) GetNumDevices() int32

func (*DeviceConfigurationProto) Marshal

func (x *DeviceConfigurationProto) Marshal() ([]byte, error)

func (*DeviceConfigurationProto) MarshalAppend

func (x *DeviceConfigurationProto) MarshalAppend(b []byte) ([]byte, error)

func (*DeviceConfigurationProto) MarshalBinary

func (x *DeviceConfigurationProto) MarshalBinary() ([]byte, error)

func (*DeviceConfigurationProto) MarshalJSON

func (x *DeviceConfigurationProto) MarshalJSON() ([]byte, error)

func (*DeviceConfigurationProto) Merge

func (*DeviceConfigurationProto) ProtoReflect

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

func (*DeviceConfigurationProto) Reset

func (x *DeviceConfigurationProto) Reset()

func (*DeviceConfigurationProto) ResetVT

func (x *DeviceConfigurationProto) ResetVT()

func (*DeviceConfigurationProto) ReturnToPool

func (x *DeviceConfigurationProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*DeviceConfigurationProto) ShallowClone

func (x *DeviceConfigurationProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*DeviceConfigurationProto) Size

func (x *DeviceConfigurationProto) Size() int

func (*DeviceConfigurationProto) Unmarshal

func (x *DeviceConfigurationProto) Unmarshal(b []byte) error

func (*DeviceConfigurationProto) UnmarshalBinary

func (x *DeviceConfigurationProto) UnmarshalBinary(data []byte) error

func (*DeviceConfigurationProto) UnmarshalJSON

func (x *DeviceConfigurationProto) UnmarshalJSON(data []byte) error

func (*DeviceConfigurationProto) UnmarshalUnsafe

func (x *DeviceConfigurationProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*DeviceConfigurationProto) Validate

func (x *DeviceConfigurationProto) Validate() error

type DeviceConfigurationProtoBuilder

type DeviceConfigurationProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewDeviceConfigurationProtoBuilder

func NewDeviceConfigurationProtoBuilder() *DeviceConfigurationProtoBuilder

func (*DeviceConfigurationProtoBuilder) Build

func (*DeviceConfigurationProtoBuilder) SetDevice

func (*DeviceConfigurationProtoBuilder) SetName

func (*DeviceConfigurationProtoBuilder) SetNumDevices

type DeviceConfigurationProtoOption

type DeviceConfigurationProtoOption func(*DeviceConfigurationProto)

func DeviceConfigurationProtoWithDevice

func DeviceConfigurationProtoWithDevice(v []string) DeviceConfigurationProtoOption

func DeviceConfigurationProtoWithName

func DeviceConfigurationProtoWithName(v string) DeviceConfigurationProtoOption

func DeviceConfigurationProtoWithNumDevices

func DeviceConfigurationProtoWithNumDevices(v int32) DeviceConfigurationProtoOption

type FunctionProto

type FunctionProto struct {
	Name           string
	Input          []string
	Output         []string
	Attribute      []string
	AttributeProto []*AttributeProto
	Node           []*NodeProto
	DocString      string
	OpsetImport    []*OperatorSetIdProto
	Domain         string
	Overload       string
	ValueInfo      []*ValueInfoProto
	MetadataProps  []*StringStringEntryProto
	// contains filtered or unexported fields
}

func FunctionProtoFromPool

func FunctionProtoFromPool() *FunctionProto

func NewFunctionProto

func NewFunctionProto(opts ...FunctionProtoOption) *FunctionProto

func (*FunctionProto) Clone

func (x *FunctionProto) Clone() proto.Message

func (*FunctionProto) Equal

func (x *FunctionProto) Equal(other *FunctionProto) bool

func (*FunctionProto) GetAttribute

func (x *FunctionProto) GetAttribute() []string

func (*FunctionProto) GetAttributeProto

func (x *FunctionProto) GetAttributeProto() []*AttributeProto

func (*FunctionProto) GetDocString

func (x *FunctionProto) GetDocString() string

func (*FunctionProto) GetDomain

func (x *FunctionProto) GetDomain() string

func (*FunctionProto) GetInput

func (x *FunctionProto) GetInput() []string

func (*FunctionProto) GetMetadataProps

func (x *FunctionProto) GetMetadataProps() []*StringStringEntryProto

func (*FunctionProto) GetName

func (x *FunctionProto) GetName() string

func (*FunctionProto) GetNode

func (x *FunctionProto) GetNode() []*NodeProto

func (*FunctionProto) GetOpsetImport

func (x *FunctionProto) GetOpsetImport() []*OperatorSetIdProto

func (*FunctionProto) GetOutput

func (x *FunctionProto) GetOutput() []string

func (*FunctionProto) GetOverload

func (x *FunctionProto) GetOverload() string

func (*FunctionProto) GetValueInfo

func (x *FunctionProto) GetValueInfo() []*ValueInfoProto

func (*FunctionProto) Marshal

func (x *FunctionProto) Marshal() ([]byte, error)

func (*FunctionProto) MarshalAppend

func (x *FunctionProto) MarshalAppend(b []byte) ([]byte, error)

func (*FunctionProto) MarshalBinary

func (x *FunctionProto) MarshalBinary() ([]byte, error)

func (*FunctionProto) MarshalJSON

func (x *FunctionProto) MarshalJSON() ([]byte, error)

func (*FunctionProto) Merge

func (x *FunctionProto) Merge(src proto.Message) error

func (*FunctionProto) ProtoReflect

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

func (*FunctionProto) Reset

func (x *FunctionProto) Reset()

func (*FunctionProto) ResetVT

func (x *FunctionProto) ResetVT()

func (*FunctionProto) ReturnToPool

func (x *FunctionProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*FunctionProto) ShallowClone

func (x *FunctionProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*FunctionProto) Size

func (x *FunctionProto) Size() int

func (*FunctionProto) Unmarshal

func (x *FunctionProto) Unmarshal(b []byte) error

func (*FunctionProto) UnmarshalBinary

func (x *FunctionProto) UnmarshalBinary(data []byte) error

func (*FunctionProto) UnmarshalJSON

func (x *FunctionProto) UnmarshalJSON(data []byte) error

func (*FunctionProto) UnmarshalUnsafe

func (x *FunctionProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*FunctionProto) Validate

func (x *FunctionProto) Validate() error

type FunctionProtoBuilder

type FunctionProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewFunctionProtoBuilder

func NewFunctionProtoBuilder() *FunctionProtoBuilder

func (*FunctionProtoBuilder) Build

func (b *FunctionProtoBuilder) Build() *FunctionProto

func (*FunctionProtoBuilder) SetAttribute

func (b *FunctionProtoBuilder) SetAttribute(v []string) *FunctionProtoBuilder

func (*FunctionProtoBuilder) SetAttributeProto

func (b *FunctionProtoBuilder) SetAttributeProto(v []*AttributeProto) *FunctionProtoBuilder

func (*FunctionProtoBuilder) SetDocString

func (b *FunctionProtoBuilder) SetDocString(v string) *FunctionProtoBuilder

func (*FunctionProtoBuilder) SetDomain

func (*FunctionProtoBuilder) SetInput

func (*FunctionProtoBuilder) SetMetadataProps

func (*FunctionProtoBuilder) SetName

func (*FunctionProtoBuilder) SetNode

func (*FunctionProtoBuilder) SetOpsetImport

func (*FunctionProtoBuilder) SetOutput

func (*FunctionProtoBuilder) SetOverload

func (*FunctionProtoBuilder) SetValueInfo

type FunctionProtoOption

type FunctionProtoOption func(*FunctionProto)

func FunctionProtoWithAttribute

func FunctionProtoWithAttribute(v []string) FunctionProtoOption

func FunctionProtoWithAttributeProto

func FunctionProtoWithAttributeProto(v []*AttributeProto) FunctionProtoOption

func FunctionProtoWithDocString

func FunctionProtoWithDocString(v string) FunctionProtoOption

func FunctionProtoWithDomain

func FunctionProtoWithDomain(v string) FunctionProtoOption

func FunctionProtoWithInput

func FunctionProtoWithInput(v []string) FunctionProtoOption

func FunctionProtoWithMetadataProps

func FunctionProtoWithMetadataProps(v []*StringStringEntryProto) FunctionProtoOption

func FunctionProtoWithName

func FunctionProtoWithName(v string) FunctionProtoOption

func FunctionProtoWithNode

func FunctionProtoWithNode(v []*NodeProto) FunctionProtoOption

func FunctionProtoWithOpsetImport

func FunctionProtoWithOpsetImport(v []*OperatorSetIdProto) FunctionProtoOption

func FunctionProtoWithOutput

func FunctionProtoWithOutput(v []string) FunctionProtoOption

func FunctionProtoWithOverload

func FunctionProtoWithOverload(v string) FunctionProtoOption

func FunctionProtoWithValueInfo

func FunctionProtoWithValueInfo(v []*ValueInfoProto) FunctionProtoOption

type GraphProto

type GraphProto struct {
	Node                   []*NodeProto
	Name                   string
	Initializer            []*TensorProto
	SparseInitializer      []*SparseTensorProto
	DocString              string
	Input                  []*ValueInfoProto
	Output                 []*ValueInfoProto
	ValueInfo              []*ValueInfoProto
	QuantizationAnnotation []*TensorAnnotation
	MetadataProps          []*StringStringEntryProto
	// contains filtered or unexported fields
}

func GraphProtoFromPool

func GraphProtoFromPool() *GraphProto

func NewGraphProto

func NewGraphProto(opts ...GraphProtoOption) *GraphProto

func (*GraphProto) Clone

func (x *GraphProto) Clone() proto.Message

func (*GraphProto) Equal

func (x *GraphProto) Equal(other *GraphProto) bool

func (*GraphProto) GetDocString

func (x *GraphProto) GetDocString() string

func (*GraphProto) GetInitializer

func (x *GraphProto) GetInitializer() []*TensorProto

func (*GraphProto) GetInput

func (x *GraphProto) GetInput() []*ValueInfoProto

func (*GraphProto) GetMetadataProps

func (x *GraphProto) GetMetadataProps() []*StringStringEntryProto

func (*GraphProto) GetName

func (x *GraphProto) GetName() string

func (*GraphProto) GetNode

func (x *GraphProto) GetNode() []*NodeProto

func (*GraphProto) GetOutput

func (x *GraphProto) GetOutput() []*ValueInfoProto

func (*GraphProto) GetQuantizationAnnotation

func (x *GraphProto) GetQuantizationAnnotation() []*TensorAnnotation

func (*GraphProto) GetSparseInitializer

func (x *GraphProto) GetSparseInitializer() []*SparseTensorProto

func (*GraphProto) GetValueInfo

func (x *GraphProto) GetValueInfo() []*ValueInfoProto

func (*GraphProto) Marshal

func (x *GraphProto) Marshal() ([]byte, error)

func (*GraphProto) MarshalAppend

func (x *GraphProto) MarshalAppend(b []byte) ([]byte, error)

func (*GraphProto) MarshalBinary

func (x *GraphProto) MarshalBinary() ([]byte, error)

func (*GraphProto) MarshalJSON

func (x *GraphProto) MarshalJSON() ([]byte, error)

func (*GraphProto) Merge

func (x *GraphProto) Merge(src proto.Message) error

func (*GraphProto) ProtoReflect

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

func (*GraphProto) Reset

func (x *GraphProto) Reset()

func (*GraphProto) ResetVT

func (x *GraphProto) ResetVT()

func (*GraphProto) ReturnToPool

func (x *GraphProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*GraphProto) ShallowClone

func (x *GraphProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*GraphProto) Size

func (x *GraphProto) Size() int

func (*GraphProto) Unmarshal

func (x *GraphProto) Unmarshal(b []byte) error

func (*GraphProto) UnmarshalBinary

func (x *GraphProto) UnmarshalBinary(data []byte) error

func (*GraphProto) UnmarshalJSON

func (x *GraphProto) UnmarshalJSON(data []byte) error

func (*GraphProto) UnmarshalUnsafe

func (x *GraphProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*GraphProto) Validate

func (x *GraphProto) Validate() error

type GraphProtoBuilder

type GraphProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewGraphProtoBuilder

func NewGraphProtoBuilder() *GraphProtoBuilder

func (*GraphProtoBuilder) Build

func (b *GraphProtoBuilder) Build() *GraphProto

func (*GraphProtoBuilder) SetDocString

func (b *GraphProtoBuilder) SetDocString(v string) *GraphProtoBuilder

func (*GraphProtoBuilder) SetInitializer

func (b *GraphProtoBuilder) SetInitializer(v []*TensorProto) *GraphProtoBuilder

func (*GraphProtoBuilder) SetInput

func (*GraphProtoBuilder) SetMetadataProps

func (b *GraphProtoBuilder) SetMetadataProps(v []*StringStringEntryProto) *GraphProtoBuilder

func (*GraphProtoBuilder) SetName

func (*GraphProtoBuilder) SetNode

func (b *GraphProtoBuilder) SetNode(v []*NodeProto) *GraphProtoBuilder

func (*GraphProtoBuilder) SetOutput

func (*GraphProtoBuilder) SetQuantizationAnnotation

func (b *GraphProtoBuilder) SetQuantizationAnnotation(v []*TensorAnnotation) *GraphProtoBuilder

func (*GraphProtoBuilder) SetSparseInitializer

func (b *GraphProtoBuilder) SetSparseInitializer(v []*SparseTensorProto) *GraphProtoBuilder

func (*GraphProtoBuilder) SetValueInfo

func (b *GraphProtoBuilder) SetValueInfo(v []*ValueInfoProto) *GraphProtoBuilder

type GraphProtoOption

type GraphProtoOption func(*GraphProto)

func GraphProtoWithDocString

func GraphProtoWithDocString(v string) GraphProtoOption

func GraphProtoWithInitializer

func GraphProtoWithInitializer(v []*TensorProto) GraphProtoOption

func GraphProtoWithInput

func GraphProtoWithInput(v []*ValueInfoProto) GraphProtoOption

func GraphProtoWithMetadataProps

func GraphProtoWithMetadataProps(v []*StringStringEntryProto) GraphProtoOption

func GraphProtoWithName

func GraphProtoWithName(v string) GraphProtoOption

func GraphProtoWithNode

func GraphProtoWithNode(v []*NodeProto) GraphProtoOption

func GraphProtoWithOutput

func GraphProtoWithOutput(v []*ValueInfoProto) GraphProtoOption

func GraphProtoWithQuantizationAnnotation

func GraphProtoWithQuantizationAnnotation(v []*TensorAnnotation) GraphProtoOption

func GraphProtoWithSparseInitializer

func GraphProtoWithSparseInitializer(v []*SparseTensorProto) GraphProtoOption

func GraphProtoWithValueInfo

func GraphProtoWithValueInfo(v []*ValueInfoProto) GraphProtoOption

type IntIntListEntryProto

type IntIntListEntryProto struct {
	Key   int64
	Value []int64
	// contains filtered or unexported fields
}

func IntIntListEntryProtoFromPool

func IntIntListEntryProtoFromPool() *IntIntListEntryProto

func NewIntIntListEntryProto

func NewIntIntListEntryProto(opts ...IntIntListEntryProtoOption) *IntIntListEntryProto

func (*IntIntListEntryProto) Clone

func (x *IntIntListEntryProto) Clone() proto.Message

func (*IntIntListEntryProto) Equal

func (*IntIntListEntryProto) GetKey

func (x *IntIntListEntryProto) GetKey() int64

func (*IntIntListEntryProto) GetValue

func (x *IntIntListEntryProto) GetValue() []int64

func (*IntIntListEntryProto) Marshal

func (x *IntIntListEntryProto) Marshal() ([]byte, error)

func (*IntIntListEntryProto) MarshalAppend

func (x *IntIntListEntryProto) MarshalAppend(b []byte) ([]byte, error)

func (*IntIntListEntryProto) MarshalBinary

func (x *IntIntListEntryProto) MarshalBinary() ([]byte, error)

func (*IntIntListEntryProto) MarshalJSON

func (x *IntIntListEntryProto) MarshalJSON() ([]byte, error)

func (*IntIntListEntryProto) Merge

func (x *IntIntListEntryProto) Merge(src proto.Message) error

func (*IntIntListEntryProto) ProtoReflect

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

func (*IntIntListEntryProto) Reset

func (x *IntIntListEntryProto) Reset()

func (*IntIntListEntryProto) ResetVT

func (x *IntIntListEntryProto) ResetVT()

func (*IntIntListEntryProto) ReturnToPool

func (x *IntIntListEntryProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*IntIntListEntryProto) ShallowClone

func (x *IntIntListEntryProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*IntIntListEntryProto) Size

func (x *IntIntListEntryProto) Size() int

func (*IntIntListEntryProto) Unmarshal

func (x *IntIntListEntryProto) Unmarshal(b []byte) error

func (*IntIntListEntryProto) UnmarshalBinary

func (x *IntIntListEntryProto) UnmarshalBinary(data []byte) error

func (*IntIntListEntryProto) UnmarshalJSON

func (x *IntIntListEntryProto) UnmarshalJSON(data []byte) error

func (*IntIntListEntryProto) UnmarshalUnsafe

func (x *IntIntListEntryProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*IntIntListEntryProto) Validate

func (x *IntIntListEntryProto) Validate() error

type IntIntListEntryProtoBuilder

type IntIntListEntryProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewIntIntListEntryProtoBuilder

func NewIntIntListEntryProtoBuilder() *IntIntListEntryProtoBuilder

func (*IntIntListEntryProtoBuilder) Build

func (*IntIntListEntryProtoBuilder) SetKey

func (*IntIntListEntryProtoBuilder) SetValue

type IntIntListEntryProtoOption

type IntIntListEntryProtoOption func(*IntIntListEntryProto)

func IntIntListEntryProtoWithKey

func IntIntListEntryProtoWithKey(v int64) IntIntListEntryProtoOption

func IntIntListEntryProtoWithValue

func IntIntListEntryProtoWithValue(v []int64) IntIntListEntryProtoOption

type ModelProto

type ModelProto struct {
	IrVersion       int64
	OpsetImport     []*OperatorSetIdProto
	ProducerName    string
	ProducerVersion string
	Domain          string
	ModelVersion    int64
	DocString       string
	Graph           *GraphProto
	MetadataProps   []*StringStringEntryProto
	TrainingInfo    []*TrainingInfoProto
	Functions       []*FunctionProto
	Configuration   []*DeviceConfigurationProto
	// contains filtered or unexported fields
}

func ModelProtoFromPool

func ModelProtoFromPool() *ModelProto

func NewModelProto

func NewModelProto(opts ...ModelProtoOption) *ModelProto

func (*ModelProto) Clone

func (x *ModelProto) Clone() proto.Message

func (*ModelProto) Equal

func (x *ModelProto) Equal(other *ModelProto) bool

func (*ModelProto) GetConfiguration

func (x *ModelProto) GetConfiguration() []*DeviceConfigurationProto

func (*ModelProto) GetDocString

func (x *ModelProto) GetDocString() string

func (*ModelProto) GetDomain

func (x *ModelProto) GetDomain() string

func (*ModelProto) GetFunctions

func (x *ModelProto) GetFunctions() []*FunctionProto

func (*ModelProto) GetGraph

func (x *ModelProto) GetGraph() *GraphProto

func (*ModelProto) GetIrVersion

func (x *ModelProto) GetIrVersion() int64

func (*ModelProto) GetMetadataProps

func (x *ModelProto) GetMetadataProps() []*StringStringEntryProto

func (*ModelProto) GetModelVersion

func (x *ModelProto) GetModelVersion() int64

func (*ModelProto) GetOpsetImport

func (x *ModelProto) GetOpsetImport() []*OperatorSetIdProto

func (*ModelProto) GetProducerName

func (x *ModelProto) GetProducerName() string

func (*ModelProto) GetProducerVersion

func (x *ModelProto) GetProducerVersion() string

func (*ModelProto) GetTrainingInfo

func (x *ModelProto) GetTrainingInfo() []*TrainingInfoProto

func (*ModelProto) Marshal

func (x *ModelProto) Marshal() ([]byte, error)

func (*ModelProto) MarshalAppend

func (x *ModelProto) MarshalAppend(b []byte) ([]byte, error)

func (*ModelProto) MarshalBinary

func (x *ModelProto) MarshalBinary() ([]byte, error)

func (*ModelProto) MarshalJSON

func (x *ModelProto) MarshalJSON() ([]byte, error)

func (*ModelProto) Merge

func (x *ModelProto) Merge(src proto.Message) error

func (*ModelProto) ProtoReflect

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

func (*ModelProto) Reset

func (x *ModelProto) Reset()

func (*ModelProto) ResetVT

func (x *ModelProto) ResetVT()

func (*ModelProto) ReturnToPool

func (x *ModelProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*ModelProto) ShallowClone

func (x *ModelProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*ModelProto) Size

func (x *ModelProto) Size() int

func (*ModelProto) Unmarshal

func (x *ModelProto) Unmarshal(b []byte) error

func (*ModelProto) UnmarshalBinary

func (x *ModelProto) UnmarshalBinary(data []byte) error

func (*ModelProto) UnmarshalJSON

func (x *ModelProto) UnmarshalJSON(data []byte) error

func (*ModelProto) UnmarshalUnsafe

func (x *ModelProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*ModelProto) Validate

func (x *ModelProto) Validate() error

type ModelProtoBuilder

type ModelProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewModelProtoBuilder

func NewModelProtoBuilder() *ModelProtoBuilder

func (*ModelProtoBuilder) Build

func (b *ModelProtoBuilder) Build() *ModelProto

func (*ModelProtoBuilder) SetConfiguration

func (*ModelProtoBuilder) SetDocString

func (b *ModelProtoBuilder) SetDocString(v string) *ModelProtoBuilder

func (*ModelProtoBuilder) SetDomain

func (b *ModelProtoBuilder) SetDomain(v string) *ModelProtoBuilder

func (*ModelProtoBuilder) SetFunctions

func (b *ModelProtoBuilder) SetFunctions(v []*FunctionProto) *ModelProtoBuilder

func (*ModelProtoBuilder) SetGraph

func (*ModelProtoBuilder) SetIrVersion

func (b *ModelProtoBuilder) SetIrVersion(v int64) *ModelProtoBuilder

func (*ModelProtoBuilder) SetMetadataProps

func (b *ModelProtoBuilder) SetMetadataProps(v []*StringStringEntryProto) *ModelProtoBuilder

func (*ModelProtoBuilder) SetModelVersion

func (b *ModelProtoBuilder) SetModelVersion(v int64) *ModelProtoBuilder

func (*ModelProtoBuilder) SetOpsetImport

func (b *ModelProtoBuilder) SetOpsetImport(v []*OperatorSetIdProto) *ModelProtoBuilder

func (*ModelProtoBuilder) SetProducerName

func (b *ModelProtoBuilder) SetProducerName(v string) *ModelProtoBuilder

func (*ModelProtoBuilder) SetProducerVersion

func (b *ModelProtoBuilder) SetProducerVersion(v string) *ModelProtoBuilder

func (*ModelProtoBuilder) SetTrainingInfo

func (b *ModelProtoBuilder) SetTrainingInfo(v []*TrainingInfoProto) *ModelProtoBuilder

type ModelProtoOption

type ModelProtoOption func(*ModelProto)

func ModelProtoWithConfiguration

func ModelProtoWithConfiguration(v []*DeviceConfigurationProto) ModelProtoOption

func ModelProtoWithDocString

func ModelProtoWithDocString(v string) ModelProtoOption

func ModelProtoWithDomain

func ModelProtoWithDomain(v string) ModelProtoOption

func ModelProtoWithFunctions

func ModelProtoWithFunctions(v []*FunctionProto) ModelProtoOption

func ModelProtoWithGraph

func ModelProtoWithGraph(v *GraphProto) ModelProtoOption

func ModelProtoWithIrVersion

func ModelProtoWithIrVersion(v int64) ModelProtoOption

func ModelProtoWithMetadataProps

func ModelProtoWithMetadataProps(v []*StringStringEntryProto) ModelProtoOption

func ModelProtoWithModelVersion

func ModelProtoWithModelVersion(v int64) ModelProtoOption

func ModelProtoWithOpsetImport

func ModelProtoWithOpsetImport(v []*OperatorSetIdProto) ModelProtoOption

func ModelProtoWithProducerName

func ModelProtoWithProducerName(v string) ModelProtoOption

func ModelProtoWithProducerVersion

func ModelProtoWithProducerVersion(v string) ModelProtoOption

func ModelProtoWithTrainingInfo

func ModelProtoWithTrainingInfo(v []*TrainingInfoProto) ModelProtoOption

type NodeDeviceConfigurationProto

type NodeDeviceConfigurationProto struct {
	ConfigurationId string
	ShardingSpec    []*ShardingSpecProto
	PipelineStage   int32
	// contains filtered or unexported fields
}

func NodeDeviceConfigurationProtoFromPool

func NodeDeviceConfigurationProtoFromPool() *NodeDeviceConfigurationProto

func (*NodeDeviceConfigurationProto) Clone

func (*NodeDeviceConfigurationProto) Equal

func (*NodeDeviceConfigurationProto) GetConfigurationId

func (x *NodeDeviceConfigurationProto) GetConfigurationId() string

func (*NodeDeviceConfigurationProto) GetPipelineStage

func (x *NodeDeviceConfigurationProto) GetPipelineStage() int32

func (*NodeDeviceConfigurationProto) GetShardingSpec

func (x *NodeDeviceConfigurationProto) GetShardingSpec() []*ShardingSpecProto

func (*NodeDeviceConfigurationProto) Marshal

func (x *NodeDeviceConfigurationProto) Marshal() ([]byte, error)

func (*NodeDeviceConfigurationProto) MarshalAppend

func (x *NodeDeviceConfigurationProto) MarshalAppend(b []byte) ([]byte, error)

func (*NodeDeviceConfigurationProto) MarshalBinary

func (x *NodeDeviceConfigurationProto) MarshalBinary() ([]byte, error)

func (*NodeDeviceConfigurationProto) MarshalJSON

func (x *NodeDeviceConfigurationProto) MarshalJSON() ([]byte, error)

func (*NodeDeviceConfigurationProto) Merge

func (*NodeDeviceConfigurationProto) ProtoReflect

func (*NodeDeviceConfigurationProto) Reset

func (x *NodeDeviceConfigurationProto) Reset()

func (*NodeDeviceConfigurationProto) ResetVT

func (x *NodeDeviceConfigurationProto) ResetVT()

func (*NodeDeviceConfigurationProto) ReturnToPool

func (x *NodeDeviceConfigurationProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*NodeDeviceConfigurationProto) ShallowClone

func (x *NodeDeviceConfigurationProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*NodeDeviceConfigurationProto) Size

func (*NodeDeviceConfigurationProto) Unmarshal

func (x *NodeDeviceConfigurationProto) Unmarshal(b []byte) error

func (*NodeDeviceConfigurationProto) UnmarshalBinary

func (x *NodeDeviceConfigurationProto) UnmarshalBinary(data []byte) error

func (*NodeDeviceConfigurationProto) UnmarshalJSON

func (x *NodeDeviceConfigurationProto) UnmarshalJSON(data []byte) error

func (*NodeDeviceConfigurationProto) UnmarshalUnsafe

func (x *NodeDeviceConfigurationProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*NodeDeviceConfigurationProto) Validate

func (x *NodeDeviceConfigurationProto) Validate() error

type NodeDeviceConfigurationProtoBuilder

type NodeDeviceConfigurationProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewNodeDeviceConfigurationProtoBuilder

func NewNodeDeviceConfigurationProtoBuilder() *NodeDeviceConfigurationProtoBuilder

func (*NodeDeviceConfigurationProtoBuilder) Build

func (*NodeDeviceConfigurationProtoBuilder) SetConfigurationId

func (*NodeDeviceConfigurationProtoBuilder) SetPipelineStage

func (*NodeDeviceConfigurationProtoBuilder) SetShardingSpec

type NodeDeviceConfigurationProtoOption

type NodeDeviceConfigurationProtoOption func(*NodeDeviceConfigurationProto)

func NodeDeviceConfigurationProtoWithConfigurationId

func NodeDeviceConfigurationProtoWithConfigurationId(v string) NodeDeviceConfigurationProtoOption

func NodeDeviceConfigurationProtoWithPipelineStage

func NodeDeviceConfigurationProtoWithPipelineStage(v int32) NodeDeviceConfigurationProtoOption

func NodeDeviceConfigurationProtoWithShardingSpec

func NodeDeviceConfigurationProtoWithShardingSpec(v []*ShardingSpecProto) NodeDeviceConfigurationProtoOption

type NodeProto

type NodeProto struct {
	Input                []string
	Output               []string
	Name                 string
	OpType               string
	Domain               string
	Overload             string
	Attribute            []*AttributeProto
	DocString            string
	MetadataProps        []*StringStringEntryProto
	DeviceConfigurations []*NodeDeviceConfigurationProto
	// contains filtered or unexported fields
}

func NewNodeProto

func NewNodeProto(opts ...NodeProtoOption) *NodeProto

func NodeProtoFromPool

func NodeProtoFromPool() *NodeProto

func (*NodeProto) Clone

func (x *NodeProto) Clone() proto.Message

func (*NodeProto) Equal

func (x *NodeProto) Equal(other *NodeProto) bool

func (*NodeProto) GetAttribute

func (x *NodeProto) GetAttribute() []*AttributeProto

func (*NodeProto) GetDeviceConfigurations

func (x *NodeProto) GetDeviceConfigurations() []*NodeDeviceConfigurationProto

func (*NodeProto) GetDocString

func (x *NodeProto) GetDocString() string

func (*NodeProto) GetDomain

func (x *NodeProto) GetDomain() string

func (*NodeProto) GetInput

func (x *NodeProto) GetInput() []string

func (*NodeProto) GetMetadataProps

func (x *NodeProto) GetMetadataProps() []*StringStringEntryProto

func (*NodeProto) GetName

func (x *NodeProto) GetName() string

func (*NodeProto) GetOpType

func (x *NodeProto) GetOpType() string

func (*NodeProto) GetOutput

func (x *NodeProto) GetOutput() []string

func (*NodeProto) GetOverload

func (x *NodeProto) GetOverload() string

func (*NodeProto) Marshal

func (x *NodeProto) Marshal() ([]byte, error)

func (*NodeProto) MarshalAppend

func (x *NodeProto) MarshalAppend(b []byte) ([]byte, error)

func (*NodeProto) MarshalBinary

func (x *NodeProto) MarshalBinary() ([]byte, error)

func (*NodeProto) MarshalJSON

func (x *NodeProto) MarshalJSON() ([]byte, error)

func (*NodeProto) Merge

func (x *NodeProto) Merge(src proto.Message) error

func (*NodeProto) ProtoReflect

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

func (*NodeProto) Reset

func (x *NodeProto) Reset()

func (*NodeProto) ResetVT

func (x *NodeProto) ResetVT()

func (*NodeProto) ReturnToPool

func (x *NodeProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*NodeProto) ShallowClone

func (x *NodeProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*NodeProto) Size

func (x *NodeProto) Size() int

func (*NodeProto) Unmarshal

func (x *NodeProto) Unmarshal(b []byte) error

func (*NodeProto) UnmarshalBinary

func (x *NodeProto) UnmarshalBinary(data []byte) error

func (*NodeProto) UnmarshalJSON

func (x *NodeProto) UnmarshalJSON(data []byte) error

func (*NodeProto) UnmarshalUnsafe

func (x *NodeProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*NodeProto) Validate

func (x *NodeProto) Validate() error

type NodeProtoBuilder

type NodeProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewNodeProtoBuilder

func NewNodeProtoBuilder() *NodeProtoBuilder

func (*NodeProtoBuilder) Build

func (b *NodeProtoBuilder) Build() *NodeProto

func (*NodeProtoBuilder) SetAttribute

func (b *NodeProtoBuilder) SetAttribute(v []*AttributeProto) *NodeProtoBuilder

func (*NodeProtoBuilder) SetDeviceConfigurations

func (b *NodeProtoBuilder) SetDeviceConfigurations(v []*NodeDeviceConfigurationProto) *NodeProtoBuilder

func (*NodeProtoBuilder) SetDocString

func (b *NodeProtoBuilder) SetDocString(v string) *NodeProtoBuilder

func (*NodeProtoBuilder) SetDomain

func (b *NodeProtoBuilder) SetDomain(v string) *NodeProtoBuilder

func (*NodeProtoBuilder) SetInput

func (b *NodeProtoBuilder) SetInput(v []string) *NodeProtoBuilder

func (*NodeProtoBuilder) SetMetadataProps

func (b *NodeProtoBuilder) SetMetadataProps(v []*StringStringEntryProto) *NodeProtoBuilder

func (*NodeProtoBuilder) SetName

func (b *NodeProtoBuilder) SetName(v string) *NodeProtoBuilder

func (*NodeProtoBuilder) SetOpType

func (b *NodeProtoBuilder) SetOpType(v string) *NodeProtoBuilder

func (*NodeProtoBuilder) SetOutput

func (b *NodeProtoBuilder) SetOutput(v []string) *NodeProtoBuilder

func (*NodeProtoBuilder) SetOverload

func (b *NodeProtoBuilder) SetOverload(v string) *NodeProtoBuilder

type NodeProtoOption

type NodeProtoOption func(*NodeProto)

func NodeProtoWithAttribute

func NodeProtoWithAttribute(v []*AttributeProto) NodeProtoOption

func NodeProtoWithDeviceConfigurations

func NodeProtoWithDeviceConfigurations(v []*NodeDeviceConfigurationProto) NodeProtoOption

func NodeProtoWithDocString

func NodeProtoWithDocString(v string) NodeProtoOption

func NodeProtoWithDomain

func NodeProtoWithDomain(v string) NodeProtoOption

func NodeProtoWithInput

func NodeProtoWithInput(v []string) NodeProtoOption

func NodeProtoWithMetadataProps

func NodeProtoWithMetadataProps(v []*StringStringEntryProto) NodeProtoOption

func NodeProtoWithName

func NodeProtoWithName(v string) NodeProtoOption

func NodeProtoWithOpType

func NodeProtoWithOpType(v string) NodeProtoOption

func NodeProtoWithOutput

func NodeProtoWithOutput(v []string) NodeProtoOption

func NodeProtoWithOverload

func NodeProtoWithOverload(v string) NodeProtoOption

type OperatorSetIdProto

type OperatorSetIdProto struct {
	Domain  string
	Version int64
	// contains filtered or unexported fields
}

func NewOperatorSetIdProto

func NewOperatorSetIdProto(opts ...OperatorSetIdProtoOption) *OperatorSetIdProto

func OperatorSetIdProtoFromPool

func OperatorSetIdProtoFromPool() *OperatorSetIdProto

func (*OperatorSetIdProto) Clone

func (x *OperatorSetIdProto) Clone() proto.Message

func (*OperatorSetIdProto) Equal

func (x *OperatorSetIdProto) Equal(other *OperatorSetIdProto) bool

func (*OperatorSetIdProto) GetDomain

func (x *OperatorSetIdProto) GetDomain() string

func (*OperatorSetIdProto) GetVersion

func (x *OperatorSetIdProto) GetVersion() int64

func (*OperatorSetIdProto) Marshal

func (x *OperatorSetIdProto) Marshal() ([]byte, error)

func (*OperatorSetIdProto) MarshalAppend

func (x *OperatorSetIdProto) MarshalAppend(b []byte) ([]byte, error)

func (*OperatorSetIdProto) MarshalBinary

func (x *OperatorSetIdProto) MarshalBinary() ([]byte, error)

func (*OperatorSetIdProto) MarshalJSON

func (x *OperatorSetIdProto) MarshalJSON() ([]byte, error)

func (*OperatorSetIdProto) Merge

func (x *OperatorSetIdProto) Merge(src proto.Message) error

func (*OperatorSetIdProto) ProtoReflect

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

func (*OperatorSetIdProto) Reset

func (x *OperatorSetIdProto) Reset()

func (*OperatorSetIdProto) ResetVT

func (x *OperatorSetIdProto) ResetVT()

func (*OperatorSetIdProto) ReturnToPool

func (x *OperatorSetIdProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*OperatorSetIdProto) ShallowClone

func (x *OperatorSetIdProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*OperatorSetIdProto) Size

func (x *OperatorSetIdProto) Size() int

func (*OperatorSetIdProto) Unmarshal

func (x *OperatorSetIdProto) Unmarshal(b []byte) error

func (*OperatorSetIdProto) UnmarshalBinary

func (x *OperatorSetIdProto) UnmarshalBinary(data []byte) error

func (*OperatorSetIdProto) UnmarshalJSON

func (x *OperatorSetIdProto) UnmarshalJSON(data []byte) error

func (*OperatorSetIdProto) UnmarshalUnsafe

func (x *OperatorSetIdProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*OperatorSetIdProto) Validate

func (x *OperatorSetIdProto) Validate() error

type OperatorSetIdProtoBuilder

type OperatorSetIdProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewOperatorSetIdProtoBuilder

func NewOperatorSetIdProtoBuilder() *OperatorSetIdProtoBuilder

func (*OperatorSetIdProtoBuilder) Build

func (*OperatorSetIdProtoBuilder) SetDomain

func (*OperatorSetIdProtoBuilder) SetVersion

type OperatorSetIdProtoOption

type OperatorSetIdProtoOption func(*OperatorSetIdProto)

func OperatorSetIdProtoWithDomain

func OperatorSetIdProtoWithDomain(v string) OperatorSetIdProtoOption

func OperatorSetIdProtoWithVersion

func OperatorSetIdProtoWithVersion(v int64) OperatorSetIdProtoOption

type OperatorStatus

type OperatorStatus int32
const (
	OperatorStatus_EXPERIMENTAL OperatorStatus = 0
	OperatorStatus_STABLE       OperatorStatus = 1
)

func ParseOperatorStatus

func ParseOperatorStatus(name string) (OperatorStatus, bool)

func (OperatorStatus) Number

func (x OperatorStatus) Number() int32

func (OperatorStatus) String

func (x OperatorStatus) String() string

type ShardedDimProto

type ShardedDimProto struct {
	Axis           int64
	SimpleSharding []*SimpleShardedDimProto
	// contains filtered or unexported fields
}

func NewShardedDimProto

func NewShardedDimProto(opts ...ShardedDimProtoOption) *ShardedDimProto

func ShardedDimProtoFromPool

func ShardedDimProtoFromPool() *ShardedDimProto

func (*ShardedDimProto) Clone

func (x *ShardedDimProto) Clone() proto.Message

func (*ShardedDimProto) Equal

func (x *ShardedDimProto) Equal(other *ShardedDimProto) bool

func (*ShardedDimProto) GetAxis

func (x *ShardedDimProto) GetAxis() int64

func (*ShardedDimProto) GetSimpleSharding

func (x *ShardedDimProto) GetSimpleSharding() []*SimpleShardedDimProto

func (*ShardedDimProto) Marshal

func (x *ShardedDimProto) Marshal() ([]byte, error)

func (*ShardedDimProto) MarshalAppend

func (x *ShardedDimProto) MarshalAppend(b []byte) ([]byte, error)

func (*ShardedDimProto) MarshalBinary

func (x *ShardedDimProto) MarshalBinary() ([]byte, error)

func (*ShardedDimProto) MarshalJSON

func (x *ShardedDimProto) MarshalJSON() ([]byte, error)

func (*ShardedDimProto) Merge

func (x *ShardedDimProto) Merge(src proto.Message) error

func (*ShardedDimProto) ProtoReflect

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

func (*ShardedDimProto) Reset

func (x *ShardedDimProto) Reset()

func (*ShardedDimProto) ResetVT

func (x *ShardedDimProto) ResetVT()

func (*ShardedDimProto) ReturnToPool

func (x *ShardedDimProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*ShardedDimProto) ShallowClone

func (x *ShardedDimProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*ShardedDimProto) Size

func (x *ShardedDimProto) Size() int

func (*ShardedDimProto) Unmarshal

func (x *ShardedDimProto) Unmarshal(b []byte) error

func (*ShardedDimProto) UnmarshalBinary

func (x *ShardedDimProto) UnmarshalBinary(data []byte) error

func (*ShardedDimProto) UnmarshalJSON

func (x *ShardedDimProto) UnmarshalJSON(data []byte) error

func (*ShardedDimProto) UnmarshalUnsafe

func (x *ShardedDimProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*ShardedDimProto) Validate

func (x *ShardedDimProto) Validate() error

type ShardedDimProtoBuilder

type ShardedDimProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewShardedDimProtoBuilder

func NewShardedDimProtoBuilder() *ShardedDimProtoBuilder

func (*ShardedDimProtoBuilder) Build

func (*ShardedDimProtoBuilder) SetAxis

func (*ShardedDimProtoBuilder) SetSimpleSharding

type ShardedDimProtoOption

type ShardedDimProtoOption func(*ShardedDimProto)

func ShardedDimProtoWithAxis

func ShardedDimProtoWithAxis(v int64) ShardedDimProtoOption

func ShardedDimProtoWithSimpleSharding

func ShardedDimProtoWithSimpleSharding(v []*SimpleShardedDimProto) ShardedDimProtoOption

type ShardingSpecProto

type ShardingSpecProto struct {
	TensorName            string
	Device                []int64
	IndexToDeviceGroupMap []*IntIntListEntryProto
	ShardedDim            []*ShardedDimProto
	// contains filtered or unexported fields
}

func NewShardingSpecProto

func NewShardingSpecProto(opts ...ShardingSpecProtoOption) *ShardingSpecProto

func ShardingSpecProtoFromPool

func ShardingSpecProtoFromPool() *ShardingSpecProto

func (*ShardingSpecProto) Clone

func (x *ShardingSpecProto) Clone() proto.Message

func (*ShardingSpecProto) Equal

func (x *ShardingSpecProto) Equal(other *ShardingSpecProto) bool

func (*ShardingSpecProto) GetDevice

func (x *ShardingSpecProto) GetDevice() []int64

func (*ShardingSpecProto) GetIndexToDeviceGroupMap

func (x *ShardingSpecProto) GetIndexToDeviceGroupMap() []*IntIntListEntryProto

func (*ShardingSpecProto) GetShardedDim

func (x *ShardingSpecProto) GetShardedDim() []*ShardedDimProto

func (*ShardingSpecProto) GetTensorName

func (x *ShardingSpecProto) GetTensorName() string

func (*ShardingSpecProto) Marshal

func (x *ShardingSpecProto) Marshal() ([]byte, error)

func (*ShardingSpecProto) MarshalAppend

func (x *ShardingSpecProto) MarshalAppend(b []byte) ([]byte, error)

func (*ShardingSpecProto) MarshalBinary

func (x *ShardingSpecProto) MarshalBinary() ([]byte, error)

func (*ShardingSpecProto) MarshalJSON

func (x *ShardingSpecProto) MarshalJSON() ([]byte, error)

func (*ShardingSpecProto) Merge

func (x *ShardingSpecProto) Merge(src proto.Message) error

func (*ShardingSpecProto) ProtoReflect

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

func (*ShardingSpecProto) Reset

func (x *ShardingSpecProto) Reset()

func (*ShardingSpecProto) ResetVT

func (x *ShardingSpecProto) ResetVT()

func (*ShardingSpecProto) ReturnToPool

func (x *ShardingSpecProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*ShardingSpecProto) ShallowClone

func (x *ShardingSpecProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*ShardingSpecProto) Size

func (x *ShardingSpecProto) Size() int

func (*ShardingSpecProto) Unmarshal

func (x *ShardingSpecProto) Unmarshal(b []byte) error

func (*ShardingSpecProto) UnmarshalBinary

func (x *ShardingSpecProto) UnmarshalBinary(data []byte) error

func (*ShardingSpecProto) UnmarshalJSON

func (x *ShardingSpecProto) UnmarshalJSON(data []byte) error

func (*ShardingSpecProto) UnmarshalUnsafe

func (x *ShardingSpecProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*ShardingSpecProto) Validate

func (x *ShardingSpecProto) Validate() error

type ShardingSpecProtoBuilder

type ShardingSpecProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewShardingSpecProtoBuilder

func NewShardingSpecProtoBuilder() *ShardingSpecProtoBuilder

func (*ShardingSpecProtoBuilder) Build

func (*ShardingSpecProtoBuilder) SetDevice

func (*ShardingSpecProtoBuilder) SetIndexToDeviceGroupMap

func (b *ShardingSpecProtoBuilder) SetIndexToDeviceGroupMap(v []*IntIntListEntryProto) *ShardingSpecProtoBuilder

func (*ShardingSpecProtoBuilder) SetShardedDim

func (*ShardingSpecProtoBuilder) SetTensorName

type ShardingSpecProtoOption

type ShardingSpecProtoOption func(*ShardingSpecProto)

func ShardingSpecProtoWithDevice

func ShardingSpecProtoWithDevice(v []int64) ShardingSpecProtoOption

func ShardingSpecProtoWithIndexToDeviceGroupMap

func ShardingSpecProtoWithIndexToDeviceGroupMap(v []*IntIntListEntryProto) ShardingSpecProtoOption

func ShardingSpecProtoWithShardedDim

func ShardingSpecProtoWithShardedDim(v []*ShardedDimProto) ShardingSpecProtoOption

func ShardingSpecProtoWithTensorName

func ShardingSpecProtoWithTensorName(v string) ShardingSpecProtoOption

type SimpleShardedDimProto

type SimpleShardedDimProto struct {
	Dim       isSimpleShardedDimProto_Dim
	NumShards int64
	// contains filtered or unexported fields
}

func NewSimpleShardedDimProto

func NewSimpleShardedDimProto(opts ...SimpleShardedDimProtoOption) *SimpleShardedDimProto

func SimpleShardedDimProtoFromPool

func SimpleShardedDimProtoFromPool() *SimpleShardedDimProto

func (*SimpleShardedDimProto) Clone

func (x *SimpleShardedDimProto) Clone() proto.Message

func (*SimpleShardedDimProto) Equal

func (*SimpleShardedDimProto) GetDim

func (x *SimpleShardedDimProto) GetDim() isSimpleShardedDimProto_Dim

func (*SimpleShardedDimProto) GetDimParam

func (x *SimpleShardedDimProto) GetDimParam() (string, bool)

func (*SimpleShardedDimProto) GetDimValue

func (x *SimpleShardedDimProto) GetDimValue() (int64, bool)

func (*SimpleShardedDimProto) GetNumShards

func (x *SimpleShardedDimProto) GetNumShards() int64

func (*SimpleShardedDimProto) Marshal

func (x *SimpleShardedDimProto) Marshal() ([]byte, error)

func (*SimpleShardedDimProto) MarshalAppend

func (x *SimpleShardedDimProto) MarshalAppend(b []byte) ([]byte, error)

func (*SimpleShardedDimProto) MarshalBinary

func (x *SimpleShardedDimProto) MarshalBinary() ([]byte, error)

func (*SimpleShardedDimProto) MarshalJSON

func (x *SimpleShardedDimProto) MarshalJSON() ([]byte, error)

func (*SimpleShardedDimProto) Merge

func (x *SimpleShardedDimProto) Merge(src proto.Message) error

func (*SimpleShardedDimProto) ProtoReflect

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

func (*SimpleShardedDimProto) Reset

func (x *SimpleShardedDimProto) Reset()

func (*SimpleShardedDimProto) ResetVT

func (x *SimpleShardedDimProto) ResetVT()

func (*SimpleShardedDimProto) ReturnToPool

func (x *SimpleShardedDimProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*SimpleShardedDimProto) SetDimParam

func (x *SimpleShardedDimProto) SetDimParam(v string)

func (*SimpleShardedDimProto) SetDimValue

func (x *SimpleShardedDimProto) SetDimValue(v int64)

func (*SimpleShardedDimProto) ShallowClone

func (x *SimpleShardedDimProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*SimpleShardedDimProto) Size

func (x *SimpleShardedDimProto) Size() int

func (*SimpleShardedDimProto) Unmarshal

func (x *SimpleShardedDimProto) Unmarshal(b []byte) error

func (*SimpleShardedDimProto) UnmarshalBinary

func (x *SimpleShardedDimProto) UnmarshalBinary(data []byte) error

func (*SimpleShardedDimProto) UnmarshalJSON

func (x *SimpleShardedDimProto) UnmarshalJSON(data []byte) error

func (*SimpleShardedDimProto) UnmarshalUnsafe

func (x *SimpleShardedDimProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*SimpleShardedDimProto) Validate

func (x *SimpleShardedDimProto) Validate() error

type SimpleShardedDimProtoBuilder

type SimpleShardedDimProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewSimpleShardedDimProtoBuilder

func NewSimpleShardedDimProtoBuilder() *SimpleShardedDimProtoBuilder

func (*SimpleShardedDimProtoBuilder) Build

func (*SimpleShardedDimProtoBuilder) SetDimParam

func (*SimpleShardedDimProtoBuilder) SetDimValue

func (*SimpleShardedDimProtoBuilder) SetNumShards

type SimpleShardedDimProtoOption

type SimpleShardedDimProtoOption func(*SimpleShardedDimProto)

func SimpleShardedDimProtoWithDimParam

func SimpleShardedDimProtoWithDimParam(v string) SimpleShardedDimProtoOption

func SimpleShardedDimProtoWithDimValue

func SimpleShardedDimProtoWithDimValue(v int64) SimpleShardedDimProtoOption

func SimpleShardedDimProtoWithNumShards

func SimpleShardedDimProtoWithNumShards(v int64) SimpleShardedDimProtoOption

type SimpleShardedDimProto_DimParam

type SimpleShardedDimProto_DimParam struct {
	DimParam string
}

type SimpleShardedDimProto_DimValue

type SimpleShardedDimProto_DimValue struct {
	DimValue int64
}

type SparseTensorProto

type SparseTensorProto struct {
	Values  *TensorProto
	Indices *TensorProto
	Dims    []int64
	// contains filtered or unexported fields
}

func NewSparseTensorProto

func NewSparseTensorProto(opts ...SparseTensorProtoOption) *SparseTensorProto

func SparseTensorProtoFromPool

func SparseTensorProtoFromPool() *SparseTensorProto

func (*SparseTensorProto) Clone

func (x *SparseTensorProto) Clone() proto.Message

func (*SparseTensorProto) Equal

func (x *SparseTensorProto) Equal(other *SparseTensorProto) bool

func (*SparseTensorProto) GetDims

func (x *SparseTensorProto) GetDims() []int64

func (*SparseTensorProto) GetIndices

func (x *SparseTensorProto) GetIndices() *TensorProto

func (*SparseTensorProto) GetValues

func (x *SparseTensorProto) GetValues() *TensorProto

func (*SparseTensorProto) Marshal

func (x *SparseTensorProto) Marshal() ([]byte, error)

func (*SparseTensorProto) MarshalAppend

func (x *SparseTensorProto) MarshalAppend(b []byte) ([]byte, error)

func (*SparseTensorProto) MarshalBinary

func (x *SparseTensorProto) MarshalBinary() ([]byte, error)

func (*SparseTensorProto) MarshalJSON

func (x *SparseTensorProto) MarshalJSON() ([]byte, error)

func (*SparseTensorProto) Merge

func (x *SparseTensorProto) Merge(src proto.Message) error

func (*SparseTensorProto) ProtoReflect

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

func (*SparseTensorProto) Reset

func (x *SparseTensorProto) Reset()

func (*SparseTensorProto) ResetVT

func (x *SparseTensorProto) ResetVT()

func (*SparseTensorProto) ReturnToPool

func (x *SparseTensorProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*SparseTensorProto) ShallowClone

func (x *SparseTensorProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*SparseTensorProto) Size

func (x *SparseTensorProto) Size() int

func (*SparseTensorProto) Unmarshal

func (x *SparseTensorProto) Unmarshal(b []byte) error

func (*SparseTensorProto) UnmarshalBinary

func (x *SparseTensorProto) UnmarshalBinary(data []byte) error

func (*SparseTensorProto) UnmarshalJSON

func (x *SparseTensorProto) UnmarshalJSON(data []byte) error

func (*SparseTensorProto) UnmarshalUnsafe

func (x *SparseTensorProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*SparseTensorProto) Validate

func (x *SparseTensorProto) Validate() error

type SparseTensorProtoBuilder

type SparseTensorProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewSparseTensorProtoBuilder

func NewSparseTensorProtoBuilder() *SparseTensorProtoBuilder

func (*SparseTensorProtoBuilder) Build

func (*SparseTensorProtoBuilder) SetDims

func (*SparseTensorProtoBuilder) SetIndices

func (*SparseTensorProtoBuilder) SetValues

type SparseTensorProtoOption

type SparseTensorProtoOption func(*SparseTensorProto)

func SparseTensorProtoWithDims

func SparseTensorProtoWithDims(v []int64) SparseTensorProtoOption

func SparseTensorProtoWithIndices

func SparseTensorProtoWithIndices(v *TensorProto) SparseTensorProtoOption

func SparseTensorProtoWithValues

func SparseTensorProtoWithValues(v *TensorProto) SparseTensorProtoOption

type StringStringEntryProto

type StringStringEntryProto struct {
	Key   string
	Value string
	// contains filtered or unexported fields
}

func StringStringEntryProtoFromPool

func StringStringEntryProtoFromPool() *StringStringEntryProto

func (*StringStringEntryProto) Clone

func (*StringStringEntryProto) Equal

func (*StringStringEntryProto) GetKey

func (x *StringStringEntryProto) GetKey() string

func (*StringStringEntryProto) GetValue

func (x *StringStringEntryProto) GetValue() string

func (*StringStringEntryProto) Marshal

func (x *StringStringEntryProto) Marshal() ([]byte, error)

func (*StringStringEntryProto) MarshalAppend

func (x *StringStringEntryProto) MarshalAppend(b []byte) ([]byte, error)

func (*StringStringEntryProto) MarshalBinary

func (x *StringStringEntryProto) MarshalBinary() ([]byte, error)

func (*StringStringEntryProto) MarshalJSON

func (x *StringStringEntryProto) MarshalJSON() ([]byte, error)

func (*StringStringEntryProto) Merge

func (x *StringStringEntryProto) Merge(src proto.Message) error

func (*StringStringEntryProto) ProtoReflect

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

func (*StringStringEntryProto) Reset

func (x *StringStringEntryProto) Reset()

func (*StringStringEntryProto) ResetVT

func (x *StringStringEntryProto) ResetVT()

func (*StringStringEntryProto) ReturnToPool

func (x *StringStringEntryProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*StringStringEntryProto) ShallowClone

func (x *StringStringEntryProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*StringStringEntryProto) Size

func (x *StringStringEntryProto) Size() int

func (*StringStringEntryProto) Unmarshal

func (x *StringStringEntryProto) Unmarshal(b []byte) error

func (*StringStringEntryProto) UnmarshalBinary

func (x *StringStringEntryProto) UnmarshalBinary(data []byte) error

func (*StringStringEntryProto) UnmarshalJSON

func (x *StringStringEntryProto) UnmarshalJSON(data []byte) error

func (*StringStringEntryProto) UnmarshalUnsafe

func (x *StringStringEntryProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*StringStringEntryProto) Validate

func (x *StringStringEntryProto) Validate() error

type StringStringEntryProtoBuilder

type StringStringEntryProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewStringStringEntryProtoBuilder

func NewStringStringEntryProtoBuilder() *StringStringEntryProtoBuilder

func (*StringStringEntryProtoBuilder) Build

func (*StringStringEntryProtoBuilder) SetKey

func (*StringStringEntryProtoBuilder) SetValue

type StringStringEntryProtoOption

type StringStringEntryProtoOption func(*StringStringEntryProto)

func StringStringEntryProtoWithKey

func StringStringEntryProtoWithKey(v string) StringStringEntryProtoOption

func StringStringEntryProtoWithValue

func StringStringEntryProtoWithValue(v string) StringStringEntryProtoOption

type TensorAnnotation

type TensorAnnotation struct {
	TensorName                string
	QuantParameterTensorNames []*StringStringEntryProto
	// contains filtered or unexported fields
}

func NewTensorAnnotation

func NewTensorAnnotation(opts ...TensorAnnotationOption) *TensorAnnotation

func TensorAnnotationFromPool

func TensorAnnotationFromPool() *TensorAnnotation

func (*TensorAnnotation) Clone

func (x *TensorAnnotation) Clone() proto.Message

func (*TensorAnnotation) Equal

func (x *TensorAnnotation) Equal(other *TensorAnnotation) bool

func (*TensorAnnotation) GetQuantParameterTensorNames

func (x *TensorAnnotation) GetQuantParameterTensorNames() []*StringStringEntryProto

func (*TensorAnnotation) GetTensorName

func (x *TensorAnnotation) GetTensorName() string

func (*TensorAnnotation) Marshal

func (x *TensorAnnotation) Marshal() ([]byte, error)

func (*TensorAnnotation) MarshalAppend

func (x *TensorAnnotation) MarshalAppend(b []byte) ([]byte, error)

func (*TensorAnnotation) MarshalBinary

func (x *TensorAnnotation) MarshalBinary() ([]byte, error)

func (*TensorAnnotation) MarshalJSON

func (x *TensorAnnotation) MarshalJSON() ([]byte, error)

func (*TensorAnnotation) Merge

func (x *TensorAnnotation) Merge(src proto.Message) error

func (*TensorAnnotation) ProtoReflect

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

func (*TensorAnnotation) Reset

func (x *TensorAnnotation) Reset()

func (*TensorAnnotation) ResetVT

func (x *TensorAnnotation) ResetVT()

func (*TensorAnnotation) ReturnToPool

func (x *TensorAnnotation) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TensorAnnotation) ShallowClone

func (x *TensorAnnotation) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TensorAnnotation) Size

func (x *TensorAnnotation) Size() int

func (*TensorAnnotation) Unmarshal

func (x *TensorAnnotation) Unmarshal(b []byte) error

func (*TensorAnnotation) UnmarshalBinary

func (x *TensorAnnotation) UnmarshalBinary(data []byte) error

func (*TensorAnnotation) UnmarshalJSON

func (x *TensorAnnotation) UnmarshalJSON(data []byte) error

func (*TensorAnnotation) UnmarshalUnsafe

func (x *TensorAnnotation) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TensorAnnotation) Validate

func (x *TensorAnnotation) Validate() error

type TensorAnnotationBuilder

type TensorAnnotationBuilder struct {
	// contains filtered or unexported fields
}

func NewTensorAnnotationBuilder

func NewTensorAnnotationBuilder() *TensorAnnotationBuilder

func (*TensorAnnotationBuilder) Build

func (*TensorAnnotationBuilder) SetQuantParameterTensorNames

func (b *TensorAnnotationBuilder) SetQuantParameterTensorNames(v []*StringStringEntryProto) *TensorAnnotationBuilder

func (*TensorAnnotationBuilder) SetTensorName

type TensorAnnotationOption

type TensorAnnotationOption func(*TensorAnnotation)

func TensorAnnotationWithQuantParameterTensorNames

func TensorAnnotationWithQuantParameterTensorNames(v []*StringStringEntryProto) TensorAnnotationOption

func TensorAnnotationWithTensorName

func TensorAnnotationWithTensorName(v string) TensorAnnotationOption

type TensorProto

type TensorProto struct {
	Dims          []int64
	DataType      int32
	Segment       *TensorProto_Segment
	FloatData     []float32
	Int32Data     []int32
	StringData    [][]byte
	Int64Data     []int64
	Name          string
	DocString     string
	RawData       []byte
	ExternalData  []*StringStringEntryProto
	DataLocation  int32
	DoubleData    []float64
	Uint64Data    []uint64
	MetadataProps []*StringStringEntryProto
	// contains filtered or unexported fields
}

func NewTensorProto

func NewTensorProto(opts ...TensorProtoOption) *TensorProto

func TensorProtoFromPool

func TensorProtoFromPool() *TensorProto

func (*TensorProto) Clone

func (x *TensorProto) Clone() proto.Message

func (*TensorProto) Equal

func (x *TensorProto) Equal(other *TensorProto) bool

func (*TensorProto) GetDataLocation

func (x *TensorProto) GetDataLocation() int32

func (*TensorProto) GetDataType

func (x *TensorProto) GetDataType() int32

func (*TensorProto) GetDims

func (x *TensorProto) GetDims() []int64

func (*TensorProto) GetDocString

func (x *TensorProto) GetDocString() string

func (*TensorProto) GetDoubleData

func (x *TensorProto) GetDoubleData() []float64

func (*TensorProto) GetExternalData

func (x *TensorProto) GetExternalData() []*StringStringEntryProto

func (*TensorProto) GetFloatData

func (x *TensorProto) GetFloatData() []float32

func (*TensorProto) GetInt32Data

func (x *TensorProto) GetInt32Data() []int32

func (*TensorProto) GetInt64Data

func (x *TensorProto) GetInt64Data() []int64

func (*TensorProto) GetMetadataProps

func (x *TensorProto) GetMetadataProps() []*StringStringEntryProto

func (*TensorProto) GetName

func (x *TensorProto) GetName() string

func (*TensorProto) GetRawData

func (x *TensorProto) GetRawData() []byte

func (*TensorProto) GetSegment

func (x *TensorProto) GetSegment() *TensorProto_Segment

func (*TensorProto) GetStringData

func (x *TensorProto) GetStringData() [][]byte

func (*TensorProto) GetUint64Data

func (x *TensorProto) GetUint64Data() []uint64

func (*TensorProto) Marshal

func (x *TensorProto) Marshal() ([]byte, error)

func (*TensorProto) MarshalAppend

func (x *TensorProto) MarshalAppend(b []byte) ([]byte, error)

func (*TensorProto) MarshalBinary

func (x *TensorProto) MarshalBinary() ([]byte, error)

func (*TensorProto) MarshalJSON

func (x *TensorProto) MarshalJSON() ([]byte, error)

func (*TensorProto) Merge

func (x *TensorProto) Merge(src proto.Message) error

func (*TensorProto) ProtoReflect

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

func (*TensorProto) Reset

func (x *TensorProto) Reset()

func (*TensorProto) ResetVT

func (x *TensorProto) ResetVT()

func (*TensorProto) ReturnToPool

func (x *TensorProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TensorProto) ShallowClone

func (x *TensorProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TensorProto) Size

func (x *TensorProto) Size() int

func (*TensorProto) Unmarshal

func (x *TensorProto) Unmarshal(b []byte) error

func (*TensorProto) UnmarshalBinary

func (x *TensorProto) UnmarshalBinary(data []byte) error

func (*TensorProto) UnmarshalJSON

func (x *TensorProto) UnmarshalJSON(data []byte) error

func (*TensorProto) UnmarshalUnsafe

func (x *TensorProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TensorProto) Validate

func (x *TensorProto) Validate() error

type TensorProtoBuilder

type TensorProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewTensorProtoBuilder

func NewTensorProtoBuilder() *TensorProtoBuilder

func (*TensorProtoBuilder) Build

func (b *TensorProtoBuilder) Build() *TensorProto

func (*TensorProtoBuilder) SetDataLocation

func (b *TensorProtoBuilder) SetDataLocation(v int32) *TensorProtoBuilder

func (*TensorProtoBuilder) SetDataType

func (b *TensorProtoBuilder) SetDataType(v int32) *TensorProtoBuilder

func (*TensorProtoBuilder) SetDims

func (b *TensorProtoBuilder) SetDims(v []int64) *TensorProtoBuilder

func (*TensorProtoBuilder) SetDocString

func (b *TensorProtoBuilder) SetDocString(v string) *TensorProtoBuilder

func (*TensorProtoBuilder) SetDoubleData

func (b *TensorProtoBuilder) SetDoubleData(v []float64) *TensorProtoBuilder

func (*TensorProtoBuilder) SetExternalData

func (*TensorProtoBuilder) SetFloatData

func (b *TensorProtoBuilder) SetFloatData(v []float32) *TensorProtoBuilder

func (*TensorProtoBuilder) SetInt32Data

func (b *TensorProtoBuilder) SetInt32Data(v []int32) *TensorProtoBuilder

func (*TensorProtoBuilder) SetInt64Data

func (b *TensorProtoBuilder) SetInt64Data(v []int64) *TensorProtoBuilder

func (*TensorProtoBuilder) SetMetadataProps

func (*TensorProtoBuilder) SetName

func (*TensorProtoBuilder) SetRawData

func (b *TensorProtoBuilder) SetRawData(v []byte) *TensorProtoBuilder

func (*TensorProtoBuilder) SetSegment

func (*TensorProtoBuilder) SetStringData

func (b *TensorProtoBuilder) SetStringData(v [][]byte) *TensorProtoBuilder

func (*TensorProtoBuilder) SetUint64Data

func (b *TensorProtoBuilder) SetUint64Data(v []uint64) *TensorProtoBuilder

type TensorProtoOption

type TensorProtoOption func(*TensorProto)

func TensorProtoWithDataLocation

func TensorProtoWithDataLocation(v int32) TensorProtoOption

func TensorProtoWithDataType

func TensorProtoWithDataType(v int32) TensorProtoOption

func TensorProtoWithDims

func TensorProtoWithDims(v []int64) TensorProtoOption

func TensorProtoWithDocString

func TensorProtoWithDocString(v string) TensorProtoOption

func TensorProtoWithDoubleData

func TensorProtoWithDoubleData(v []float64) TensorProtoOption

func TensorProtoWithExternalData

func TensorProtoWithExternalData(v []*StringStringEntryProto) TensorProtoOption

func TensorProtoWithFloatData

func TensorProtoWithFloatData(v []float32) TensorProtoOption

func TensorProtoWithInt32Data

func TensorProtoWithInt32Data(v []int32) TensorProtoOption

func TensorProtoWithInt64Data

func TensorProtoWithInt64Data(v []int64) TensorProtoOption

func TensorProtoWithMetadataProps

func TensorProtoWithMetadataProps(v []*StringStringEntryProto) TensorProtoOption

func TensorProtoWithName

func TensorProtoWithName(v string) TensorProtoOption

func TensorProtoWithRawData

func TensorProtoWithRawData(v []byte) TensorProtoOption

func TensorProtoWithSegment

func TensorProtoWithSegment(v *TensorProto_Segment) TensorProtoOption

func TensorProtoWithStringData

func TensorProtoWithStringData(v [][]byte) TensorProtoOption

func TensorProtoWithUint64Data

func TensorProtoWithUint64Data(v []uint64) TensorProtoOption

type TensorProto_DataLocation

type TensorProto_DataLocation int32
const (
	TensorProto_DataLocation_DEFAULT  TensorProto_DataLocation = 0
	TensorProto_DataLocation_EXTERNAL TensorProto_DataLocation = 1
)

func ParseTensorProto_DataLocation

func ParseTensorProto_DataLocation(name string) (TensorProto_DataLocation, bool)

func (TensorProto_DataLocation) Number

func (x TensorProto_DataLocation) Number() int32

func (TensorProto_DataLocation) String

func (x TensorProto_DataLocation) String() string

type TensorProto_DataType

type TensorProto_DataType int32
const (
	TensorProto_DataType_UNDEFINED      TensorProto_DataType = 0
	TensorProto_DataType_FLOAT          TensorProto_DataType = 1
	TensorProto_DataType_UINT8          TensorProto_DataType = 2
	TensorProto_DataType_INT8           TensorProto_DataType = 3
	TensorProto_DataType_UINT16         TensorProto_DataType = 4
	TensorProto_DataType_INT16          TensorProto_DataType = 5
	TensorProto_DataType_INT32          TensorProto_DataType = 6
	TensorProto_DataType_INT64          TensorProto_DataType = 7
	TensorProto_DataType_STRING         TensorProto_DataType = 8
	TensorProto_DataType_BOOL           TensorProto_DataType = 9
	TensorProto_DataType_FLOAT16        TensorProto_DataType = 10
	TensorProto_DataType_DOUBLE         TensorProto_DataType = 11
	TensorProto_DataType_UINT32         TensorProto_DataType = 12
	TensorProto_DataType_UINT64         TensorProto_DataType = 13
	TensorProto_DataType_COMPLEX64      TensorProto_DataType = 14
	TensorProto_DataType_COMPLEX128     TensorProto_DataType = 15
	TensorProto_DataType_BFLOAT16       TensorProto_DataType = 16
	TensorProto_DataType_FLOAT8E4M3FN   TensorProto_DataType = 17
	TensorProto_DataType_FLOAT8E4M3FNUZ TensorProto_DataType = 18
	TensorProto_DataType_FLOAT8E5M2     TensorProto_DataType = 19
	TensorProto_DataType_FLOAT8E5M2FNUZ TensorProto_DataType = 20
	TensorProto_DataType_UINT4          TensorProto_DataType = 21
	TensorProto_DataType_INT4           TensorProto_DataType = 22
	TensorProto_DataType_FLOAT4E2M1     TensorProto_DataType = 23
	TensorProto_DataType_FLOAT8E8M0     TensorProto_DataType = 24
	TensorProto_DataType_UINT2          TensorProto_DataType = 25
	TensorProto_DataType_INT2           TensorProto_DataType = 26
)

func ParseTensorProto_DataType

func ParseTensorProto_DataType(name string) (TensorProto_DataType, bool)

func (TensorProto_DataType) Number

func (x TensorProto_DataType) Number() int32

func (TensorProto_DataType) String

func (x TensorProto_DataType) String() string

type TensorProto_Segment

type TensorProto_Segment struct {
	Begin int64
	End   int64
	// contains filtered or unexported fields
}

func NewTensorProto_Segment

func NewTensorProto_Segment(opts ...TensorProto_SegmentOption) *TensorProto_Segment

func TensorProto_SegmentFromPool

func TensorProto_SegmentFromPool() *TensorProto_Segment

func (*TensorProto_Segment) Clone

func (x *TensorProto_Segment) Clone() proto.Message

func (*TensorProto_Segment) Equal

func (x *TensorProto_Segment) Equal(other *TensorProto_Segment) bool

func (*TensorProto_Segment) GetBegin

func (x *TensorProto_Segment) GetBegin() int64

func (*TensorProto_Segment) GetEnd

func (x *TensorProto_Segment) GetEnd() int64

func (*TensorProto_Segment) Marshal

func (x *TensorProto_Segment) Marshal() ([]byte, error)

func (*TensorProto_Segment) MarshalAppend

func (x *TensorProto_Segment) MarshalAppend(b []byte) ([]byte, error)

func (*TensorProto_Segment) MarshalBinary

func (x *TensorProto_Segment) MarshalBinary() ([]byte, error)

func (*TensorProto_Segment) MarshalJSON

func (x *TensorProto_Segment) MarshalJSON() ([]byte, error)

func (*TensorProto_Segment) Merge

func (x *TensorProto_Segment) Merge(src proto.Message) error

func (*TensorProto_Segment) ProtoReflect

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

func (*TensorProto_Segment) Reset

func (x *TensorProto_Segment) Reset()

func (*TensorProto_Segment) ResetVT

func (x *TensorProto_Segment) ResetVT()

func (*TensorProto_Segment) ReturnToPool

func (x *TensorProto_Segment) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TensorProto_Segment) ShallowClone

func (x *TensorProto_Segment) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TensorProto_Segment) Size

func (x *TensorProto_Segment) Size() int

func (*TensorProto_Segment) Unmarshal

func (x *TensorProto_Segment) Unmarshal(b []byte) error

func (*TensorProto_Segment) UnmarshalBinary

func (x *TensorProto_Segment) UnmarshalBinary(data []byte) error

func (*TensorProto_Segment) UnmarshalJSON

func (x *TensorProto_Segment) UnmarshalJSON(data []byte) error

func (*TensorProto_Segment) UnmarshalUnsafe

func (x *TensorProto_Segment) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TensorProto_Segment) Validate

func (x *TensorProto_Segment) Validate() error

type TensorProto_SegmentBuilder

type TensorProto_SegmentBuilder struct {
	// contains filtered or unexported fields
}

func NewTensorProto_SegmentBuilder

func NewTensorProto_SegmentBuilder() *TensorProto_SegmentBuilder

func (*TensorProto_SegmentBuilder) Build

func (*TensorProto_SegmentBuilder) SetBegin

func (*TensorProto_SegmentBuilder) SetEnd

type TensorProto_SegmentOption

type TensorProto_SegmentOption func(*TensorProto_Segment)

func TensorProto_SegmentWithBegin

func TensorProto_SegmentWithBegin(v int64) TensorProto_SegmentOption

func TensorProto_SegmentWithEnd

func TensorProto_SegmentWithEnd(v int64) TensorProto_SegmentOption

type TensorShapeProto

type TensorShapeProto struct {
	Dim []*TensorShapeProto_Dimension
	// contains filtered or unexported fields
}

func NewTensorShapeProto

func NewTensorShapeProto(opts ...TensorShapeProtoOption) *TensorShapeProto

func TensorShapeProtoFromPool

func TensorShapeProtoFromPool() *TensorShapeProto

func (*TensorShapeProto) Clone

func (x *TensorShapeProto) Clone() proto.Message

func (*TensorShapeProto) Equal

func (x *TensorShapeProto) Equal(other *TensorShapeProto) bool

func (*TensorShapeProto) GetDim

func (*TensorShapeProto) Marshal

func (x *TensorShapeProto) Marshal() ([]byte, error)

func (*TensorShapeProto) MarshalAppend

func (x *TensorShapeProto) MarshalAppend(b []byte) ([]byte, error)

func (*TensorShapeProto) MarshalBinary

func (x *TensorShapeProto) MarshalBinary() ([]byte, error)

func (*TensorShapeProto) MarshalJSON

func (x *TensorShapeProto) MarshalJSON() ([]byte, error)

func (*TensorShapeProto) Merge

func (x *TensorShapeProto) Merge(src proto.Message) error

func (*TensorShapeProto) ProtoReflect

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

func (*TensorShapeProto) Reset

func (x *TensorShapeProto) Reset()

func (*TensorShapeProto) ResetVT

func (x *TensorShapeProto) ResetVT()

func (*TensorShapeProto) ReturnToPool

func (x *TensorShapeProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TensorShapeProto) ShallowClone

func (x *TensorShapeProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TensorShapeProto) Size

func (x *TensorShapeProto) Size() int

func (*TensorShapeProto) Unmarshal

func (x *TensorShapeProto) Unmarshal(b []byte) error

func (*TensorShapeProto) UnmarshalBinary

func (x *TensorShapeProto) UnmarshalBinary(data []byte) error

func (*TensorShapeProto) UnmarshalJSON

func (x *TensorShapeProto) UnmarshalJSON(data []byte) error

func (*TensorShapeProto) UnmarshalUnsafe

func (x *TensorShapeProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TensorShapeProto) Validate

func (x *TensorShapeProto) Validate() error

type TensorShapeProtoBuilder

type TensorShapeProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewTensorShapeProtoBuilder

func NewTensorShapeProtoBuilder() *TensorShapeProtoBuilder

func (*TensorShapeProtoBuilder) Build

func (*TensorShapeProtoBuilder) SetDim

type TensorShapeProtoOption

type TensorShapeProtoOption func(*TensorShapeProto)

type TensorShapeProto_Dimension

type TensorShapeProto_Dimension struct {
	Value      isTensorShapeProto_Dimension_Value
	Denotation string
	// contains filtered or unexported fields
}

func TensorShapeProto_DimensionFromPool

func TensorShapeProto_DimensionFromPool() *TensorShapeProto_Dimension

func (*TensorShapeProto_Dimension) Clone

func (*TensorShapeProto_Dimension) Equal

func (*TensorShapeProto_Dimension) GetDenotation

func (x *TensorShapeProto_Dimension) GetDenotation() string

func (*TensorShapeProto_Dimension) GetDimParam

func (x *TensorShapeProto_Dimension) GetDimParam() (string, bool)

func (*TensorShapeProto_Dimension) GetDimValue

func (x *TensorShapeProto_Dimension) GetDimValue() (int64, bool)

func (*TensorShapeProto_Dimension) GetValue

func (x *TensorShapeProto_Dimension) GetValue() isTensorShapeProto_Dimension_Value

func (*TensorShapeProto_Dimension) Marshal

func (x *TensorShapeProto_Dimension) Marshal() ([]byte, error)

func (*TensorShapeProto_Dimension) MarshalAppend

func (x *TensorShapeProto_Dimension) MarshalAppend(b []byte) ([]byte, error)

func (*TensorShapeProto_Dimension) MarshalBinary

func (x *TensorShapeProto_Dimension) MarshalBinary() ([]byte, error)

func (*TensorShapeProto_Dimension) MarshalJSON

func (x *TensorShapeProto_Dimension) MarshalJSON() ([]byte, error)

func (*TensorShapeProto_Dimension) Merge

func (*TensorShapeProto_Dimension) ProtoReflect

func (*TensorShapeProto_Dimension) Reset

func (x *TensorShapeProto_Dimension) Reset()

func (*TensorShapeProto_Dimension) ResetVT

func (x *TensorShapeProto_Dimension) ResetVT()

func (*TensorShapeProto_Dimension) ReturnToPool

func (x *TensorShapeProto_Dimension) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TensorShapeProto_Dimension) SetDimParam

func (x *TensorShapeProto_Dimension) SetDimParam(v string)

func (*TensorShapeProto_Dimension) SetDimValue

func (x *TensorShapeProto_Dimension) SetDimValue(v int64)

func (*TensorShapeProto_Dimension) ShallowClone

func (x *TensorShapeProto_Dimension) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TensorShapeProto_Dimension) Size

func (x *TensorShapeProto_Dimension) Size() int

func (*TensorShapeProto_Dimension) Unmarshal

func (x *TensorShapeProto_Dimension) Unmarshal(b []byte) error

func (*TensorShapeProto_Dimension) UnmarshalBinary

func (x *TensorShapeProto_Dimension) UnmarshalBinary(data []byte) error

func (*TensorShapeProto_Dimension) UnmarshalJSON

func (x *TensorShapeProto_Dimension) UnmarshalJSON(data []byte) error

func (*TensorShapeProto_Dimension) UnmarshalUnsafe

func (x *TensorShapeProto_Dimension) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TensorShapeProto_Dimension) Validate

func (x *TensorShapeProto_Dimension) Validate() error

type TensorShapeProto_DimensionBuilder

type TensorShapeProto_DimensionBuilder struct {
	// contains filtered or unexported fields
}

func NewTensorShapeProto_DimensionBuilder

func NewTensorShapeProto_DimensionBuilder() *TensorShapeProto_DimensionBuilder

func (*TensorShapeProto_DimensionBuilder) Build

func (*TensorShapeProto_DimensionBuilder) SetDenotation

func (*TensorShapeProto_DimensionBuilder) SetDimParam

func (*TensorShapeProto_DimensionBuilder) SetDimValue

type TensorShapeProto_DimensionOption

type TensorShapeProto_DimensionOption func(*TensorShapeProto_Dimension)

func TensorShapeProto_DimensionWithDenotation

func TensorShapeProto_DimensionWithDenotation(v string) TensorShapeProto_DimensionOption

func TensorShapeProto_DimensionWithDimParam

func TensorShapeProto_DimensionWithDimParam(v string) TensorShapeProto_DimensionOption

func TensorShapeProto_DimensionWithDimValue

func TensorShapeProto_DimensionWithDimValue(v int64) TensorShapeProto_DimensionOption

type TensorShapeProto_Dimension_DimParam

type TensorShapeProto_Dimension_DimParam struct {
	DimParam string
}

type TensorShapeProto_Dimension_DimValue

type TensorShapeProto_Dimension_DimValue struct {
	DimValue int64
}

type TrainingInfoProto

type TrainingInfoProto struct {
	Initialization        *GraphProto
	Algorithm             *GraphProto
	InitializationBinding []*StringStringEntryProto
	UpdateBinding         []*StringStringEntryProto
	// contains filtered or unexported fields
}

func NewTrainingInfoProto

func NewTrainingInfoProto(opts ...TrainingInfoProtoOption) *TrainingInfoProto

func TrainingInfoProtoFromPool

func TrainingInfoProtoFromPool() *TrainingInfoProto

func (*TrainingInfoProto) Clone

func (x *TrainingInfoProto) Clone() proto.Message

func (*TrainingInfoProto) Equal

func (x *TrainingInfoProto) Equal(other *TrainingInfoProto) bool

func (*TrainingInfoProto) GetAlgorithm

func (x *TrainingInfoProto) GetAlgorithm() *GraphProto

func (*TrainingInfoProto) GetInitialization

func (x *TrainingInfoProto) GetInitialization() *GraphProto

func (*TrainingInfoProto) GetInitializationBinding

func (x *TrainingInfoProto) GetInitializationBinding() []*StringStringEntryProto

func (*TrainingInfoProto) GetUpdateBinding

func (x *TrainingInfoProto) GetUpdateBinding() []*StringStringEntryProto

func (*TrainingInfoProto) Marshal

func (x *TrainingInfoProto) Marshal() ([]byte, error)

func (*TrainingInfoProto) MarshalAppend

func (x *TrainingInfoProto) MarshalAppend(b []byte) ([]byte, error)

func (*TrainingInfoProto) MarshalBinary

func (x *TrainingInfoProto) MarshalBinary() ([]byte, error)

func (*TrainingInfoProto) MarshalJSON

func (x *TrainingInfoProto) MarshalJSON() ([]byte, error)

func (*TrainingInfoProto) Merge

func (x *TrainingInfoProto) Merge(src proto.Message) error

func (*TrainingInfoProto) ProtoReflect

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

func (*TrainingInfoProto) Reset

func (x *TrainingInfoProto) Reset()

func (*TrainingInfoProto) ResetVT

func (x *TrainingInfoProto) ResetVT()

func (*TrainingInfoProto) ReturnToPool

func (x *TrainingInfoProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TrainingInfoProto) ShallowClone

func (x *TrainingInfoProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TrainingInfoProto) Size

func (x *TrainingInfoProto) Size() int

func (*TrainingInfoProto) Unmarshal

func (x *TrainingInfoProto) Unmarshal(b []byte) error

func (*TrainingInfoProto) UnmarshalBinary

func (x *TrainingInfoProto) UnmarshalBinary(data []byte) error

func (*TrainingInfoProto) UnmarshalJSON

func (x *TrainingInfoProto) UnmarshalJSON(data []byte) error

func (*TrainingInfoProto) UnmarshalUnsafe

func (x *TrainingInfoProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TrainingInfoProto) Validate

func (x *TrainingInfoProto) Validate() error

type TrainingInfoProtoBuilder

type TrainingInfoProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewTrainingInfoProtoBuilder

func NewTrainingInfoProtoBuilder() *TrainingInfoProtoBuilder

func (*TrainingInfoProtoBuilder) Build

func (*TrainingInfoProtoBuilder) SetAlgorithm

func (*TrainingInfoProtoBuilder) SetInitialization

func (*TrainingInfoProtoBuilder) SetInitializationBinding

func (*TrainingInfoProtoBuilder) SetUpdateBinding

type TrainingInfoProtoOption

type TrainingInfoProtoOption func(*TrainingInfoProto)

func TrainingInfoProtoWithAlgorithm

func TrainingInfoProtoWithAlgorithm(v *GraphProto) TrainingInfoProtoOption

func TrainingInfoProtoWithInitialization

func TrainingInfoProtoWithInitialization(v *GraphProto) TrainingInfoProtoOption

func TrainingInfoProtoWithInitializationBinding

func TrainingInfoProtoWithInitializationBinding(v []*StringStringEntryProto) TrainingInfoProtoOption

func TrainingInfoProtoWithUpdateBinding

func TrainingInfoProtoWithUpdateBinding(v []*StringStringEntryProto) TrainingInfoProtoOption

type TypeProto

type TypeProto struct {
	Value      isTypeProto_Value
	Denotation string
	// contains filtered or unexported fields
}

func NewTypeProto

func NewTypeProto(opts ...TypeProtoOption) *TypeProto

func TypeProtoFromPool

func TypeProtoFromPool() *TypeProto

func (*TypeProto) Clone

func (x *TypeProto) Clone() proto.Message

func (*TypeProto) Equal

func (x *TypeProto) Equal(other *TypeProto) bool

func (*TypeProto) GetDenotation

func (x *TypeProto) GetDenotation() string

func (*TypeProto) GetMapType

func (x *TypeProto) GetMapType() (*TypeProto_Map, bool)

func (*TypeProto) GetOptionalType

func (x *TypeProto) GetOptionalType() (*TypeProto_Optional, bool)

func (*TypeProto) GetSequenceType

func (x *TypeProto) GetSequenceType() (*TypeProto_Sequence, bool)

func (*TypeProto) GetSparseTensorType

func (x *TypeProto) GetSparseTensorType() (*TypeProto_SparseTensor, bool)

func (*TypeProto) GetTensorType

func (x *TypeProto) GetTensorType() (*TypeProto_Tensor, bool)

func (*TypeProto) GetValue

func (x *TypeProto) GetValue() isTypeProto_Value

func (*TypeProto) Marshal

func (x *TypeProto) Marshal() ([]byte, error)

func (*TypeProto) MarshalAppend

func (x *TypeProto) MarshalAppend(b []byte) ([]byte, error)

func (*TypeProto) MarshalBinary

func (x *TypeProto) MarshalBinary() ([]byte, error)

func (*TypeProto) MarshalJSON

func (x *TypeProto) MarshalJSON() ([]byte, error)

func (*TypeProto) Merge

func (x *TypeProto) Merge(src proto.Message) error

func (*TypeProto) ProtoReflect

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

func (*TypeProto) Reset

func (x *TypeProto) Reset()

func (*TypeProto) ResetVT

func (x *TypeProto) ResetVT()

func (*TypeProto) ReturnToPool

func (x *TypeProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TypeProto) SetMapType

func (x *TypeProto) SetMapType(v *TypeProto_Map)

func (*TypeProto) SetOptionalType

func (x *TypeProto) SetOptionalType(v *TypeProto_Optional)

func (*TypeProto) SetSequenceType

func (x *TypeProto) SetSequenceType(v *TypeProto_Sequence)

func (*TypeProto) SetSparseTensorType

func (x *TypeProto) SetSparseTensorType(v *TypeProto_SparseTensor)

func (*TypeProto) SetTensorType

func (x *TypeProto) SetTensorType(v *TypeProto_Tensor)

func (*TypeProto) ShallowClone

func (x *TypeProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TypeProto) Size

func (x *TypeProto) Size() int

func (*TypeProto) Unmarshal

func (x *TypeProto) Unmarshal(b []byte) error

func (*TypeProto) UnmarshalBinary

func (x *TypeProto) UnmarshalBinary(data []byte) error

func (*TypeProto) UnmarshalJSON

func (x *TypeProto) UnmarshalJSON(data []byte) error

func (*TypeProto) UnmarshalUnsafe

func (x *TypeProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TypeProto) Validate

func (x *TypeProto) Validate() error

type TypeProtoBuilder

type TypeProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewTypeProtoBuilder

func NewTypeProtoBuilder() *TypeProtoBuilder

func (*TypeProtoBuilder) Build

func (b *TypeProtoBuilder) Build() *TypeProto

func (*TypeProtoBuilder) SetDenotation

func (b *TypeProtoBuilder) SetDenotation(v string) *TypeProtoBuilder

func (*TypeProtoBuilder) SetMapType

func (b *TypeProtoBuilder) SetMapType(v *TypeProto_Map) *TypeProtoBuilder

func (*TypeProtoBuilder) SetOptionalType

func (b *TypeProtoBuilder) SetOptionalType(v *TypeProto_Optional) *TypeProtoBuilder

func (*TypeProtoBuilder) SetSequenceType

func (b *TypeProtoBuilder) SetSequenceType(v *TypeProto_Sequence) *TypeProtoBuilder

func (*TypeProtoBuilder) SetSparseTensorType

func (b *TypeProtoBuilder) SetSparseTensorType(v *TypeProto_SparseTensor) *TypeProtoBuilder

func (*TypeProtoBuilder) SetTensorType

func (b *TypeProtoBuilder) SetTensorType(v *TypeProto_Tensor) *TypeProtoBuilder

type TypeProtoOption

type TypeProtoOption func(*TypeProto)

func TypeProtoWithDenotation

func TypeProtoWithDenotation(v string) TypeProtoOption

func TypeProtoWithMapType

func TypeProtoWithMapType(v *TypeProto_Map) TypeProtoOption

func TypeProtoWithOptionalType

func TypeProtoWithOptionalType(v *TypeProto_Optional) TypeProtoOption

func TypeProtoWithSequenceType

func TypeProtoWithSequenceType(v *TypeProto_Sequence) TypeProtoOption

func TypeProtoWithSparseTensorType

func TypeProtoWithSparseTensorType(v *TypeProto_SparseTensor) TypeProtoOption

func TypeProtoWithTensorType

func TypeProtoWithTensorType(v *TypeProto_Tensor) TypeProtoOption

type TypeProto_Map

type TypeProto_Map struct {
	KeyType   int32
	ValueType *TypeProto
	// contains filtered or unexported fields
}

func NewTypeProto_Map

func NewTypeProto_Map(opts ...TypeProto_MapOption) *TypeProto_Map

func TypeProto_MapFromPool

func TypeProto_MapFromPool() *TypeProto_Map

func (*TypeProto_Map) Clone

func (x *TypeProto_Map) Clone() proto.Message

func (*TypeProto_Map) Equal

func (x *TypeProto_Map) Equal(other *TypeProto_Map) bool

func (*TypeProto_Map) GetKeyType

func (x *TypeProto_Map) GetKeyType() int32

func (*TypeProto_Map) GetValueType

func (x *TypeProto_Map) GetValueType() *TypeProto

func (*TypeProto_Map) Marshal

func (x *TypeProto_Map) Marshal() ([]byte, error)

func (*TypeProto_Map) MarshalAppend

func (x *TypeProto_Map) MarshalAppend(b []byte) ([]byte, error)

func (*TypeProto_Map) MarshalBinary

func (x *TypeProto_Map) MarshalBinary() ([]byte, error)

func (*TypeProto_Map) MarshalJSON

func (x *TypeProto_Map) MarshalJSON() ([]byte, error)

func (*TypeProto_Map) Merge

func (x *TypeProto_Map) Merge(src proto.Message) error

func (*TypeProto_Map) ProtoReflect

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

func (*TypeProto_Map) Reset

func (x *TypeProto_Map) Reset()

func (*TypeProto_Map) ResetVT

func (x *TypeProto_Map) ResetVT()

func (*TypeProto_Map) ReturnToPool

func (x *TypeProto_Map) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TypeProto_Map) ShallowClone

func (x *TypeProto_Map) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TypeProto_Map) Size

func (x *TypeProto_Map) Size() int

func (*TypeProto_Map) Unmarshal

func (x *TypeProto_Map) Unmarshal(b []byte) error

func (*TypeProto_Map) UnmarshalBinary

func (x *TypeProto_Map) UnmarshalBinary(data []byte) error

func (*TypeProto_Map) UnmarshalJSON

func (x *TypeProto_Map) UnmarshalJSON(data []byte) error

func (*TypeProto_Map) UnmarshalUnsafe

func (x *TypeProto_Map) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TypeProto_Map) Validate

func (x *TypeProto_Map) Validate() error

type TypeProto_MapBuilder

type TypeProto_MapBuilder struct {
	// contains filtered or unexported fields
}

func NewTypeProto_MapBuilder

func NewTypeProto_MapBuilder() *TypeProto_MapBuilder

func (*TypeProto_MapBuilder) Build

func (b *TypeProto_MapBuilder) Build() *TypeProto_Map

func (*TypeProto_MapBuilder) SetKeyType

func (*TypeProto_MapBuilder) SetValueType

type TypeProto_MapOption

type TypeProto_MapOption func(*TypeProto_Map)

func TypeProto_MapWithKeyType

func TypeProto_MapWithKeyType(v int32) TypeProto_MapOption

func TypeProto_MapWithValueType

func TypeProto_MapWithValueType(v *TypeProto) TypeProto_MapOption

type TypeProto_MapType

type TypeProto_MapType struct {
	MapType *TypeProto_Map
}

type TypeProto_Optional

type TypeProto_Optional struct {
	ElemType *TypeProto
	// contains filtered or unexported fields
}

func NewTypeProto_Optional

func NewTypeProto_Optional(opts ...TypeProto_OptionalOption) *TypeProto_Optional

func TypeProto_OptionalFromPool

func TypeProto_OptionalFromPool() *TypeProto_Optional

func (*TypeProto_Optional) Clone

func (x *TypeProto_Optional) Clone() proto.Message

func (*TypeProto_Optional) Equal

func (x *TypeProto_Optional) Equal(other *TypeProto_Optional) bool

func (*TypeProto_Optional) GetElemType

func (x *TypeProto_Optional) GetElemType() *TypeProto

func (*TypeProto_Optional) Marshal

func (x *TypeProto_Optional) Marshal() ([]byte, error)

func (*TypeProto_Optional) MarshalAppend

func (x *TypeProto_Optional) MarshalAppend(b []byte) ([]byte, error)

func (*TypeProto_Optional) MarshalBinary

func (x *TypeProto_Optional) MarshalBinary() ([]byte, error)

func (*TypeProto_Optional) MarshalJSON

func (x *TypeProto_Optional) MarshalJSON() ([]byte, error)

func (*TypeProto_Optional) Merge

func (x *TypeProto_Optional) Merge(src proto.Message) error

func (*TypeProto_Optional) ProtoReflect

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

func (*TypeProto_Optional) Reset

func (x *TypeProto_Optional) Reset()

func (*TypeProto_Optional) ResetVT

func (x *TypeProto_Optional) ResetVT()

func (*TypeProto_Optional) ReturnToPool

func (x *TypeProto_Optional) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TypeProto_Optional) ShallowClone

func (x *TypeProto_Optional) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TypeProto_Optional) Size

func (x *TypeProto_Optional) Size() int

func (*TypeProto_Optional) Unmarshal

func (x *TypeProto_Optional) Unmarshal(b []byte) error

func (*TypeProto_Optional) UnmarshalBinary

func (x *TypeProto_Optional) UnmarshalBinary(data []byte) error

func (*TypeProto_Optional) UnmarshalJSON

func (x *TypeProto_Optional) UnmarshalJSON(data []byte) error

func (*TypeProto_Optional) UnmarshalUnsafe

func (x *TypeProto_Optional) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TypeProto_Optional) Validate

func (x *TypeProto_Optional) Validate() error

type TypeProto_OptionalBuilder

type TypeProto_OptionalBuilder struct {
	// contains filtered or unexported fields
}

func NewTypeProto_OptionalBuilder

func NewTypeProto_OptionalBuilder() *TypeProto_OptionalBuilder

func (*TypeProto_OptionalBuilder) Build

func (*TypeProto_OptionalBuilder) SetElemType

type TypeProto_OptionalOption

type TypeProto_OptionalOption func(*TypeProto_Optional)

func TypeProto_OptionalWithElemType

func TypeProto_OptionalWithElemType(v *TypeProto) TypeProto_OptionalOption

type TypeProto_OptionalType

type TypeProto_OptionalType struct {
	OptionalType *TypeProto_Optional
}

type TypeProto_Sequence

type TypeProto_Sequence struct {
	ElemType *TypeProto
	// contains filtered or unexported fields
}

func NewTypeProto_Sequence

func NewTypeProto_Sequence(opts ...TypeProto_SequenceOption) *TypeProto_Sequence

func TypeProto_SequenceFromPool

func TypeProto_SequenceFromPool() *TypeProto_Sequence

func (*TypeProto_Sequence) Clone

func (x *TypeProto_Sequence) Clone() proto.Message

func (*TypeProto_Sequence) Equal

func (x *TypeProto_Sequence) Equal(other *TypeProto_Sequence) bool

func (*TypeProto_Sequence) GetElemType

func (x *TypeProto_Sequence) GetElemType() *TypeProto

func (*TypeProto_Sequence) Marshal

func (x *TypeProto_Sequence) Marshal() ([]byte, error)

func (*TypeProto_Sequence) MarshalAppend

func (x *TypeProto_Sequence) MarshalAppend(b []byte) ([]byte, error)

func (*TypeProto_Sequence) MarshalBinary

func (x *TypeProto_Sequence) MarshalBinary() ([]byte, error)

func (*TypeProto_Sequence) MarshalJSON

func (x *TypeProto_Sequence) MarshalJSON() ([]byte, error)

func (*TypeProto_Sequence) Merge

func (x *TypeProto_Sequence) Merge(src proto.Message) error

func (*TypeProto_Sequence) ProtoReflect

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

func (*TypeProto_Sequence) Reset

func (x *TypeProto_Sequence) Reset()

func (*TypeProto_Sequence) ResetVT

func (x *TypeProto_Sequence) ResetVT()

func (*TypeProto_Sequence) ReturnToPool

func (x *TypeProto_Sequence) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TypeProto_Sequence) ShallowClone

func (x *TypeProto_Sequence) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TypeProto_Sequence) Size

func (x *TypeProto_Sequence) Size() int

func (*TypeProto_Sequence) Unmarshal

func (x *TypeProto_Sequence) Unmarshal(b []byte) error

func (*TypeProto_Sequence) UnmarshalBinary

func (x *TypeProto_Sequence) UnmarshalBinary(data []byte) error

func (*TypeProto_Sequence) UnmarshalJSON

func (x *TypeProto_Sequence) UnmarshalJSON(data []byte) error

func (*TypeProto_Sequence) UnmarshalUnsafe

func (x *TypeProto_Sequence) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TypeProto_Sequence) Validate

func (x *TypeProto_Sequence) Validate() error

type TypeProto_SequenceBuilder

type TypeProto_SequenceBuilder struct {
	// contains filtered or unexported fields
}

func NewTypeProto_SequenceBuilder

func NewTypeProto_SequenceBuilder() *TypeProto_SequenceBuilder

func (*TypeProto_SequenceBuilder) Build

func (*TypeProto_SequenceBuilder) SetElemType

type TypeProto_SequenceOption

type TypeProto_SequenceOption func(*TypeProto_Sequence)

func TypeProto_SequenceWithElemType

func TypeProto_SequenceWithElemType(v *TypeProto) TypeProto_SequenceOption

type TypeProto_SequenceType

type TypeProto_SequenceType struct {
	SequenceType *TypeProto_Sequence
}

type TypeProto_SparseTensor

type TypeProto_SparseTensor struct {
	ElemType int32
	Shape    *TensorShapeProto
	// contains filtered or unexported fields
}

func TypeProto_SparseTensorFromPool

func TypeProto_SparseTensorFromPool() *TypeProto_SparseTensor

func (*TypeProto_SparseTensor) Clone

func (*TypeProto_SparseTensor) Equal

func (*TypeProto_SparseTensor) GetElemType

func (x *TypeProto_SparseTensor) GetElemType() int32

func (*TypeProto_SparseTensor) GetShape

func (*TypeProto_SparseTensor) Marshal

func (x *TypeProto_SparseTensor) Marshal() ([]byte, error)

func (*TypeProto_SparseTensor) MarshalAppend

func (x *TypeProto_SparseTensor) MarshalAppend(b []byte) ([]byte, error)

func (*TypeProto_SparseTensor) MarshalBinary

func (x *TypeProto_SparseTensor) MarshalBinary() ([]byte, error)

func (*TypeProto_SparseTensor) MarshalJSON

func (x *TypeProto_SparseTensor) MarshalJSON() ([]byte, error)

func (*TypeProto_SparseTensor) Merge

func (x *TypeProto_SparseTensor) Merge(src proto.Message) error

func (*TypeProto_SparseTensor) ProtoReflect

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

func (*TypeProto_SparseTensor) Reset

func (x *TypeProto_SparseTensor) Reset()

func (*TypeProto_SparseTensor) ResetVT

func (x *TypeProto_SparseTensor) ResetVT()

func (*TypeProto_SparseTensor) ReturnToPool

func (x *TypeProto_SparseTensor) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TypeProto_SparseTensor) ShallowClone

func (x *TypeProto_SparseTensor) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TypeProto_SparseTensor) Size

func (x *TypeProto_SparseTensor) Size() int

func (*TypeProto_SparseTensor) Unmarshal

func (x *TypeProto_SparseTensor) Unmarshal(b []byte) error

func (*TypeProto_SparseTensor) UnmarshalBinary

func (x *TypeProto_SparseTensor) UnmarshalBinary(data []byte) error

func (*TypeProto_SparseTensor) UnmarshalJSON

func (x *TypeProto_SparseTensor) UnmarshalJSON(data []byte) error

func (*TypeProto_SparseTensor) UnmarshalUnsafe

func (x *TypeProto_SparseTensor) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TypeProto_SparseTensor) Validate

func (x *TypeProto_SparseTensor) Validate() error

type TypeProto_SparseTensorBuilder

type TypeProto_SparseTensorBuilder struct {
	// contains filtered or unexported fields
}

func NewTypeProto_SparseTensorBuilder

func NewTypeProto_SparseTensorBuilder() *TypeProto_SparseTensorBuilder

func (*TypeProto_SparseTensorBuilder) Build

func (*TypeProto_SparseTensorBuilder) SetElemType

func (*TypeProto_SparseTensorBuilder) SetShape

type TypeProto_SparseTensorOption

type TypeProto_SparseTensorOption func(*TypeProto_SparseTensor)

func TypeProto_SparseTensorWithElemType

func TypeProto_SparseTensorWithElemType(v int32) TypeProto_SparseTensorOption

type TypeProto_SparseTensorType

type TypeProto_SparseTensorType struct {
	SparseTensorType *TypeProto_SparseTensor
}

type TypeProto_Tensor

type TypeProto_Tensor struct {
	ElemType int32
	Shape    *TensorShapeProto
	// contains filtered or unexported fields
}

func NewTypeProto_Tensor

func NewTypeProto_Tensor(opts ...TypeProto_TensorOption) *TypeProto_Tensor

func TypeProto_TensorFromPool

func TypeProto_TensorFromPool() *TypeProto_Tensor

func (*TypeProto_Tensor) Clone

func (x *TypeProto_Tensor) Clone() proto.Message

func (*TypeProto_Tensor) Equal

func (x *TypeProto_Tensor) Equal(other *TypeProto_Tensor) bool

func (*TypeProto_Tensor) GetElemType

func (x *TypeProto_Tensor) GetElemType() int32

func (*TypeProto_Tensor) GetShape

func (x *TypeProto_Tensor) GetShape() *TensorShapeProto

func (*TypeProto_Tensor) Marshal

func (x *TypeProto_Tensor) Marshal() ([]byte, error)

func (*TypeProto_Tensor) MarshalAppend

func (x *TypeProto_Tensor) MarshalAppend(b []byte) ([]byte, error)

func (*TypeProto_Tensor) MarshalBinary

func (x *TypeProto_Tensor) MarshalBinary() ([]byte, error)

func (*TypeProto_Tensor) MarshalJSON

func (x *TypeProto_Tensor) MarshalJSON() ([]byte, error)

func (*TypeProto_Tensor) Merge

func (x *TypeProto_Tensor) Merge(src proto.Message) error

func (*TypeProto_Tensor) ProtoReflect

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

func (*TypeProto_Tensor) Reset

func (x *TypeProto_Tensor) Reset()

func (*TypeProto_Tensor) ResetVT

func (x *TypeProto_Tensor) ResetVT()

func (*TypeProto_Tensor) ReturnToPool

func (x *TypeProto_Tensor) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*TypeProto_Tensor) ShallowClone

func (x *TypeProto_Tensor) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*TypeProto_Tensor) Size

func (x *TypeProto_Tensor) Size() int

func (*TypeProto_Tensor) Unmarshal

func (x *TypeProto_Tensor) Unmarshal(b []byte) error

func (*TypeProto_Tensor) UnmarshalBinary

func (x *TypeProto_Tensor) UnmarshalBinary(data []byte) error

func (*TypeProto_Tensor) UnmarshalJSON

func (x *TypeProto_Tensor) UnmarshalJSON(data []byte) error

func (*TypeProto_Tensor) UnmarshalUnsafe

func (x *TypeProto_Tensor) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*TypeProto_Tensor) Validate

func (x *TypeProto_Tensor) Validate() error

type TypeProto_TensorBuilder

type TypeProto_TensorBuilder struct {
	// contains filtered or unexported fields
}

func NewTypeProto_TensorBuilder

func NewTypeProto_TensorBuilder() *TypeProto_TensorBuilder

func (*TypeProto_TensorBuilder) Build

func (*TypeProto_TensorBuilder) SetElemType

func (*TypeProto_TensorBuilder) SetShape

type TypeProto_TensorOption

type TypeProto_TensorOption func(*TypeProto_Tensor)

func TypeProto_TensorWithElemType

func TypeProto_TensorWithElemType(v int32) TypeProto_TensorOption

func TypeProto_TensorWithShape

func TypeProto_TensorWithShape(v *TensorShapeProto) TypeProto_TensorOption

type TypeProto_TensorType

type TypeProto_TensorType struct {
	TensorType *TypeProto_Tensor
}

type ValueInfoProto

type ValueInfoProto struct {
	Name          string
	Type          *TypeProto
	DocString     string
	MetadataProps []*StringStringEntryProto
	// contains filtered or unexported fields
}

func NewValueInfoProto

func NewValueInfoProto(opts ...ValueInfoProtoOption) *ValueInfoProto

func ValueInfoProtoFromPool

func ValueInfoProtoFromPool() *ValueInfoProto

func (*ValueInfoProto) Clone

func (x *ValueInfoProto) Clone() proto.Message

func (*ValueInfoProto) Equal

func (x *ValueInfoProto) Equal(other *ValueInfoProto) bool

func (*ValueInfoProto) GetDocString

func (x *ValueInfoProto) GetDocString() string

func (*ValueInfoProto) GetMetadataProps

func (x *ValueInfoProto) GetMetadataProps() []*StringStringEntryProto

func (*ValueInfoProto) GetName

func (x *ValueInfoProto) GetName() string

func (*ValueInfoProto) GetType

func (x *ValueInfoProto) GetType() *TypeProto

func (*ValueInfoProto) Marshal

func (x *ValueInfoProto) Marshal() ([]byte, error)

func (*ValueInfoProto) MarshalAppend

func (x *ValueInfoProto) MarshalAppend(b []byte) ([]byte, error)

func (*ValueInfoProto) MarshalBinary

func (x *ValueInfoProto) MarshalBinary() ([]byte, error)

func (*ValueInfoProto) MarshalJSON

func (x *ValueInfoProto) MarshalJSON() ([]byte, error)

func (*ValueInfoProto) Merge

func (x *ValueInfoProto) Merge(src proto.Message) error

func (*ValueInfoProto) ProtoReflect

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

func (*ValueInfoProto) Reset

func (x *ValueInfoProto) Reset()

func (*ValueInfoProto) ResetVT

func (x *ValueInfoProto) ResetVT()

func (*ValueInfoProto) ReturnToPool

func (x *ValueInfoProto) ReturnToPool()

ReturnToPool resets and returns this message to the pool. The message must not be used after calling ReturnToPool.

func (*ValueInfoProto) ShallowClone

func (x *ValueInfoProto) ShallowClone() proto.Message

ShallowClone returns a shallow copy of the message. Scalar fields are copied by value. Slice, map, and message pointer fields copy their references without deep copying; mutations to shared slices, maps, or sub-messages will affect both the original and the clone.

func (*ValueInfoProto) Size

func (x *ValueInfoProto) Size() int

func (*ValueInfoProto) Unmarshal

func (x *ValueInfoProto) Unmarshal(b []byte) error

func (*ValueInfoProto) UnmarshalBinary

func (x *ValueInfoProto) UnmarshalBinary(data []byte) error

func (*ValueInfoProto) UnmarshalJSON

func (x *ValueInfoProto) UnmarshalJSON(data []byte) error

func (*ValueInfoProto) UnmarshalUnsafe

func (x *ValueInfoProto) UnmarshalUnsafe(b []byte, opts proto.UnsafeUnmarshalOptions) error

UnmarshalUnsafe decodes wire bytes with optional zero-copy string aliasing. When opts.ZeroCopyStrings is true, string and bytes fields alias the input buffer directly. The caller MUST guarantee the input []byte outlives the message and is not modified while the message is in use. Violation of this contract causes undefined behavior.

func (*ValueInfoProto) Validate

func (x *ValueInfoProto) Validate() error

type ValueInfoProtoBuilder

type ValueInfoProtoBuilder struct {
	// contains filtered or unexported fields
}

func NewValueInfoProtoBuilder

func NewValueInfoProtoBuilder() *ValueInfoProtoBuilder

func (*ValueInfoProtoBuilder) Build

func (*ValueInfoProtoBuilder) SetDocString

func (*ValueInfoProtoBuilder) SetMetadataProps

func (*ValueInfoProtoBuilder) SetName

func (*ValueInfoProtoBuilder) SetType

type ValueInfoProtoOption

type ValueInfoProtoOption func(*ValueInfoProto)

func ValueInfoProtoWithDocString

func ValueInfoProtoWithDocString(v string) ValueInfoProtoOption

func ValueInfoProtoWithMetadataProps

func ValueInfoProtoWithMetadataProps(v []*StringStringEntryProto) ValueInfoProtoOption

func ValueInfoProtoWithName

func ValueInfoProtoWithName(v string) ValueInfoProtoOption

func ValueInfoProtoWithType

func ValueInfoProtoWithType(v *TypeProto) ValueInfoProtoOption

type Version

type Version int32
const (
	Version__START_VERSION        Version = 0
	Version_IR_VERSION_2017_10_10 Version = 1
	Version_IR_VERSION_2017_10_30 Version = 2
	Version_IR_VERSION_2017_11_3  Version = 3
	Version_IR_VERSION_2019_1_22  Version = 4
	Version_IR_VERSION_2019_3_18  Version = 5
	Version_IR_VERSION_2019_9_19  Version = 6
	Version_IR_VERSION_2020_5_8   Version = 7
	Version_IR_VERSION_2021_7_30  Version = 8
	Version_IR_VERSION_2023_5_5   Version = 9
	Version_IR_VERSION_2024_3_25  Version = 10
	Version_IR_VERSION_2025_05_12 Version = 11
	Version_IR_VERSION_2025_08_26 Version = 12
	Version_IR_VERSION            Version = 13
)

func ParseVersion

func ParseVersion(name string) (Version, bool)

func (Version) Number

func (x Version) Number() int32

func (Version) String

func (x Version) String() string

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL