data

package

Go to main page

Versions in this module

v2

v2.14.1

Dec 8, 2023

v2.14.0

Dec 8, 2023

Changes in this version

type LiteralProto

+ F8E4M3B11Fnuzs []byte

+ F8E4M3Fnuzs []byte

+ F8E5M2Fnuzs []byte

+ func (x *LiteralProto) GetF8E4M3B11Fnuzs() []byte

+ func (x *LiteralProto) GetF8E4M3Fnuzs() []byte

+ func (x *LiteralProto) GetF8E5M2Fnuzs() []byte

type OpMetadata

+ PreserveLayout bool

+ StackFrameId int32

+ func (x *OpMetadata) GetPreserveLayout() bool

+ func (x *OpMetadata) GetStackFrameId() int32

type OpSharding

+ IotaReshapeDims []int64

+ IotaTransposePerm []int32

+ func (x *OpSharding) GetIotaReshapeDims() []int64

+ func (x *OpSharding) GetIotaTransposePerm() []int32

type PrimitiveType

+ const PrimitiveType_F8E4M3B11FNUZ

+ const PrimitiveType_F8E4M3FNUZ

+ const PrimitiveType_F8E5M2FNUZ

+ type Statistic struct

+ StatName string

+ StatVal float64

+ func (*Statistic) Descriptor() ([]byte, []int)

+ func (*Statistic) ProtoMessage()

+ func (x *Statistic) GetStatName() string

+ func (x *Statistic) GetStatVal() float64

+ func (x *Statistic) ProtoReflect() protoreflect.Message

+ func (x *Statistic) Reset()

+ func (x *Statistic) String() string

+ type StatisticsViz struct

+ StatIndexToVisualize int64

+ Statistics []*Statistic

+ func (*StatisticsViz) Descriptor() ([]byte, []int)

+ func (*StatisticsViz) ProtoMessage()

+ func (x *StatisticsViz) GetStatIndexToVisualize() int64

+ func (x *StatisticsViz) GetStatistics() []*Statistic

+ func (x *StatisticsViz) ProtoReflect() protoreflect.Message

+ func (x *StatisticsViz) Reset()

+ func (x *StatisticsViz) String() string

v2.13.0

Dec 8, 2023

Changes in this version

type DimLevelType

+ const DimLevelType_DIM_COMPRESSED_WITH_HI

type LiteralProto

+ S4S []byte

+ U4S []byte

+ func (x *LiteralProto) GetS4S() []byte

+ func (x *LiteralProto) GetU4S() []byte

type OpMetadata

+ DeduplicatedName string

+ func (x *OpMetadata) GetDeduplicatedName() string

type PrimitiveType

+ const PrimitiveType_S4

+ const PrimitiveType_U4

v2.12.0

Dec 8, 2023

Changes in this version

type LayoutProto

+ DimOrdered []bool

+ DimUnique []bool

+ DynamicShapeMetadataPrefixBytes int64

+ IndexPrimitiveType PrimitiveType

+ PointerPrimitiveType PrimitiveType

+ func (x *LayoutProto) GetDimOrdered() []bool

+ func (x *LayoutProto) GetDimUnique() []bool

+ func (x *LayoutProto) GetDynamicShapeMetadataPrefixBytes() int64

+ func (x *LayoutProto) GetIndexPrimitiveType() PrimitiveType

+ func (x *LayoutProto) GetPointerPrimitiveType() PrimitiveType

type LiteralProto

+ F8E4M3Fns []byte

+ F8E5M2S []byte

+ func (x *LiteralProto) GetF8E4M3Fns() []byte

+ func (x *LiteralProto) GetF8E5M2S() []byte

+ type OutputOperandAliasing struct

+ OperandIndex int64

+ OperandShapeIndex []int64

+ OutputShapeIndex []int64

+ func (*OutputOperandAliasing) Descriptor() ([]byte, []int)

+ func (*OutputOperandAliasing) ProtoMessage()

+ func (x *OutputOperandAliasing) GetOperandIndex() int64

+ func (x *OutputOperandAliasing) GetOperandShapeIndex() []int64

+ func (x *OutputOperandAliasing) GetOutputShapeIndex() []int64

+ func (x *OutputOperandAliasing) ProtoReflect() protoreflect.Message

+ func (x *OutputOperandAliasing) Reset()

+ func (x *OutputOperandAliasing) String() string

type PrimitiveType

+ const PrimitiveType_F8E4M3FN

+ const PrimitiveType_F8E5M2

v2.11.0

Dec 8, 2023

Changes in this version

+ var DimLevelType_name = map[int32]string

+ var DimLevelType_value = map[string]int32

+ type DimLevelType int32

+ const DimLevelType_DIM_COMPRESSED

+ const DimLevelType_DIM_DENSE

+ const DimLevelType_DIM_SINGLETON

+ func (DimLevelType) Descriptor() protoreflect.EnumDescriptor

+ func (DimLevelType) EnumDescriptor() ([]byte, []int)

+ func (DimLevelType) Type() protoreflect.EnumType

+ func (x DimLevelType) Enum() *DimLevelType

+ func (x DimLevelType) Number() protoreflect.EnumNumber

+ func (x DimLevelType) String() string

type LayoutProto

+ DimLevelTypes []DimLevelType

+ PhysicalShape *ShapeProto

+ func (x *LayoutProto) GetDimLevelTypes() []DimLevelType

+ func (x *LayoutProto) GetPhysicalShape() *ShapeProto

type PrecisionConfig_Precision

+ const PrecisionConfig_PACKED_NIBBLE

v2.10.1

Dec 8, 2023

v2.10.0

Dec 8, 2023

Changes in this version

+ var ChannelHandle_ChannelType_name = map[int32]string

+ var ChannelHandle_ChannelType_value = map[string]int32

+ var CompilationEvent_name = map[int32]string

+ var CompilationEvent_value = map[string]int32

+ var FftType_name = map[int32]string

+ var FftType_value = map[string]int32

+ var File_tensorflow_compiler_xla_xla_data_proto protoreflect.FileDescriptor

+ var Format_name = map[int32]string

+ var Format_value = map[string]int32

+ var OpSharding_Type_name = map[int32]string

+ var OpSharding_Type_value = map[string]int32

+ var PaddingType_name = map[int32]string

+ var PaddingType_value = map[string]int32

+ var PrecisionConfig_Precision_name = map[int32]string

+ var PrecisionConfig_Precision_value = map[string]int32

+ var PrimitiveType_name = map[int32]string

+ var PrimitiveType_value = map[string]int32

+ var ProfileSource_name = map[int32]string

+ var ProfileSource_value = map[string]int32

+ var ProfileType_name = map[int32]string

+ var ProfileType_value = map[string]int32

+ var RandomAlgorithm_name = map[int32]string

+ var RandomAlgorithm_value = map[string]int32

+ var RandomDistribution_name = map[int32]string

+ var RandomDistribution_value = map[string]int32

+ var TriangularSolveOptions_Transpose_name = map[int32]string

+ var TriangularSolveOptions_Transpose_value = map[string]int32

+ type ChannelHandle struct

+ Handle int64

+ Type ChannelHandle_ChannelType

+ func (*ChannelHandle) Descriptor() ([]byte, []int)

+ func (*ChannelHandle) ProtoMessage()

+ func (x *ChannelHandle) GetHandle() int64

+ func (x *ChannelHandle) GetType() ChannelHandle_ChannelType

+ func (x *ChannelHandle) ProtoReflect() protoreflect.Message

+ func (x *ChannelHandle) Reset()

+ func (x *ChannelHandle) String() string

+ type ChannelHandle_ChannelType int32

+ const ChannelHandle_CHANNEL_TYPE_INVALID

+ const ChannelHandle_DEVICE_TO_DEVICE

+ const ChannelHandle_DEVICE_TO_HOST

+ const ChannelHandle_HOST_TO_DEVICE

+ func (ChannelHandle_ChannelType) Descriptor() protoreflect.EnumDescriptor

+ func (ChannelHandle_ChannelType) EnumDescriptor() ([]byte, []int)

+ func (ChannelHandle_ChannelType) Type() protoreflect.EnumType

+ func (x ChannelHandle_ChannelType) Enum() *ChannelHandle_ChannelType

+ func (x ChannelHandle_ChannelType) Number() protoreflect.EnumNumber

+ func (x ChannelHandle_ChannelType) String() string

+ type CholeskyOptions struct

+ Lower bool

+ func (*CholeskyOptions) Descriptor() ([]byte, []int)

+ func (*CholeskyOptions) ProtoMessage()

+ func (x *CholeskyOptions) GetLower() bool

+ func (x *CholeskyOptions) ProtoReflect() protoreflect.Message

+ func (x *CholeskyOptions) Reset()

+ func (x *CholeskyOptions) String() string

+ type CompilationEvent int32

+ const CompilationEvent_COMPILATION_EVENT_FIRST_COMPILATION

+ const CompilationEvent_COMPILATION_EVENT_RECOMPILATION

+ const CompilationEvent_COMPILATION_EVENT_UNKNOWN_EVENT

+ func (CompilationEvent) Descriptor() protoreflect.EnumDescriptor

+ func (CompilationEvent) EnumDescriptor() ([]byte, []int)

+ func (CompilationEvent) Type() protoreflect.EnumType

+ func (x CompilationEvent) Enum() *CompilationEvent

+ func (x CompilationEvent) Number() protoreflect.EnumNumber

+ func (x CompilationEvent) String() string

+ type ComputationStats struct

+ FlopCount float64

+ TranscendentalCount float64

+ func (*ComputationStats) Descriptor() ([]byte, []int)

+ func (*ComputationStats) ProtoMessage()

+ func (x *ComputationStats) GetFlopCount() float64

+ func (x *ComputationStats) GetTranscendentalCount() float64

+ func (x *ComputationStats) ProtoReflect() protoreflect.Message

+ func (x *ComputationStats) Reset()

+ func (x *ComputationStats) String() string

+ type ConvolutionDimensionNumbers struct

+ InputBatchDimension int64

+ InputFeatureDimension int64

+ InputSpatialDimensions []int64

+ KernelInputFeatureDimension int64

+ KernelOutputFeatureDimension int64

+ KernelSpatialDimensions []int64

+ OutputBatchDimension int64

+ OutputFeatureDimension int64

+ OutputSpatialDimensions []int64

+ func (*ConvolutionDimensionNumbers) Descriptor() ([]byte, []int)

+ func (*ConvolutionDimensionNumbers) ProtoMessage()

+ func (x *ConvolutionDimensionNumbers) GetInputBatchDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetInputFeatureDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetInputSpatialDimensions() []int64

+ func (x *ConvolutionDimensionNumbers) GetKernelInputFeatureDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetKernelOutputFeatureDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetKernelSpatialDimensions() []int64

+ func (x *ConvolutionDimensionNumbers) GetOutputBatchDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetOutputFeatureDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetOutputSpatialDimensions() []int64

+ func (x *ConvolutionDimensionNumbers) ProtoReflect() protoreflect.Message

+ func (x *ConvolutionDimensionNumbers) Reset()

+ func (x *ConvolutionDimensionNumbers) String() string

+ type CustomCallOutputOperandAliasing struct

+ OperandIndex int64

+ OperandShapeIndex []int64

+ OutputShapeIndex []int64

+ func (*CustomCallOutputOperandAliasing) Descriptor() ([]byte, []int)

+ func (*CustomCallOutputOperandAliasing) ProtoMessage()

+ func (x *CustomCallOutputOperandAliasing) GetOperandIndex() int64

+ func (x *CustomCallOutputOperandAliasing) GetOperandShapeIndex() []int64

+ func (x *CustomCallOutputOperandAliasing) GetOutputShapeIndex() []int64

+ func (x *CustomCallOutputOperandAliasing) ProtoReflect() protoreflect.Message

+ func (x *CustomCallOutputOperandAliasing) Reset()

+ func (x *CustomCallOutputOperandAliasing) String() string

+ type DeviceAssignmentProto struct

+ ComputationCount int32

+ ComputationDevices []*DeviceAssignmentProto_ComputationDevice

+ ReplicaCount int32

+ func (*DeviceAssignmentProto) Descriptor() ([]byte, []int)

+ func (*DeviceAssignmentProto) ProtoMessage()

+ func (x *DeviceAssignmentProto) GetComputationCount() int32

+ func (x *DeviceAssignmentProto) GetComputationDevices() []*DeviceAssignmentProto_ComputationDevice

+ func (x *DeviceAssignmentProto) GetReplicaCount() int32

+ func (x *DeviceAssignmentProto) ProtoReflect() protoreflect.Message

+ func (x *DeviceAssignmentProto) Reset()

+ func (x *DeviceAssignmentProto) String() string

+ type DeviceAssignmentProto_ComputationDevice struct

+ ReplicaDeviceIds []int32

+ func (*DeviceAssignmentProto_ComputationDevice) Descriptor() ([]byte, []int)

+ func (*DeviceAssignmentProto_ComputationDevice) ProtoMessage()

+ func (x *DeviceAssignmentProto_ComputationDevice) GetReplicaDeviceIds() []int32

+ func (x *DeviceAssignmentProto_ComputationDevice) ProtoReflect() protoreflect.Message

+ func (x *DeviceAssignmentProto_ComputationDevice) Reset()

+ func (x *DeviceAssignmentProto_ComputationDevice) String() string

+ type DeviceHandle struct

+ DeviceCount int64

+ Handle int64

+ func (*DeviceHandle) Descriptor() ([]byte, []int)

+ func (*DeviceHandle) ProtoMessage()

+ func (x *DeviceHandle) GetDeviceCount() int64

+ func (x *DeviceHandle) GetHandle() int64

+ func (x *DeviceHandle) ProtoReflect() protoreflect.Message

+ func (x *DeviceHandle) Reset()

+ func (x *DeviceHandle) String() string

+ type DotDimensionNumbers struct

+ LhsBatchDimensions []int64

+ LhsContractingDimensions []int64

+ RhsBatchDimensions []int64

+ RhsContractingDimensions []int64

+ func (*DotDimensionNumbers) Descriptor() ([]byte, []int)

+ func (*DotDimensionNumbers) ProtoMessage()

+ func (x *DotDimensionNumbers) GetLhsBatchDimensions() []int64

+ func (x *DotDimensionNumbers) GetLhsContractingDimensions() []int64

+ func (x *DotDimensionNumbers) GetRhsBatchDimensions() []int64

+ func (x *DotDimensionNumbers) GetRhsContractingDimensions() []int64

+ func (x *DotDimensionNumbers) ProtoReflect() protoreflect.Message

+ func (x *DotDimensionNumbers) Reset()

+ func (x *DotDimensionNumbers) String() string

+ type ExecutionHandle struct

+ Handle int64

+ func (*ExecutionHandle) Descriptor() ([]byte, []int)

+ func (*ExecutionHandle) ProtoMessage()

+ func (x *ExecutionHandle) GetHandle() int64

+ func (x *ExecutionHandle) ProtoReflect() protoreflect.Message

+ func (x *ExecutionHandle) Reset()

+ func (x *ExecutionHandle) String() string

+ type ExecutionProfile struct

+ CompilationCacheHit bool

+ CompileTimeMs int64

+ ComputeAndTransferTimeNs int64

+ ComputeCycleCount int64

+ ComputeTimeNs int64

+ ExecutableSizeInBytes int64

+ ProfileCacheHit bool

+ func (*ExecutionProfile) Descriptor() ([]byte, []int)

+ func (*ExecutionProfile) ProtoMessage()

+ func (x *ExecutionProfile) GetCompilationCacheHit() bool

+ func (x *ExecutionProfile) GetCompileTimeMs() int64

+ func (x *ExecutionProfile) GetComputeAndTransferTimeNs() int64

+ func (x *ExecutionProfile) GetComputeCycleCount() int64

+ func (x *ExecutionProfile) GetComputeTimeNs() int64

+ func (x *ExecutionProfile) GetExecutableSizeInBytes() int64

+ func (x *ExecutionProfile) GetProfileCacheHit() bool

+ func (x *ExecutionProfile) ProtoReflect() protoreflect.Message

+ func (x *ExecutionProfile) Reset()

+ func (x *ExecutionProfile) String() string

+ type FftType int32

+ const FftType_FFT

+ const FftType_IFFT

+ const FftType_IRFFT

+ const FftType_RFFT

+ func (FftType) Descriptor() protoreflect.EnumDescriptor

+ func (FftType) EnumDescriptor() ([]byte, []int)

+ func (FftType) Type() protoreflect.EnumType

+ func (x FftType) Enum() *FftType

+ func (x FftType) Number() protoreflect.EnumNumber

+ func (x FftType) String() string

+ type Format int32

+ const Format_DENSE

+ const Format_INVALID_FORMAT

+ func (Format) Descriptor() protoreflect.EnumDescriptor

+ func (Format) EnumDescriptor() ([]byte, []int)

+ func (Format) Type() protoreflect.EnumType

+ func (x Format) Enum() *Format

+ func (x Format) Number() protoreflect.EnumNumber

+ func (x Format) String() string

+ type FrontendAttributes struct

+ Map map[string]string

+ func (*FrontendAttributes) Descriptor() ([]byte, []int)

+ func (*FrontendAttributes) ProtoMessage()

+ func (x *FrontendAttributes) GetMap() map[string]string

+ func (x *FrontendAttributes) ProtoReflect() protoreflect.Message

+ func (x *FrontendAttributes) Reset()

+ func (x *FrontendAttributes) String() string

+ type GatherDimensionNumbers struct

+ CollapsedSliceDims []int64

+ IndexVectorDim int64

+ OffsetDims []int64

+ StartIndexMap []int64

+ func (*GatherDimensionNumbers) Descriptor() ([]byte, []int)

+ func (*GatherDimensionNumbers) ProtoMessage()

+ func (x *GatherDimensionNumbers) GetCollapsedSliceDims() []int64

+ func (x *GatherDimensionNumbers) GetIndexVectorDim() int64

+ func (x *GatherDimensionNumbers) GetOffsetDims() []int64

+ func (x *GatherDimensionNumbers) GetStartIndexMap() []int64

+ func (x *GatherDimensionNumbers) ProtoReflect() protoreflect.Message

+ func (x *GatherDimensionNumbers) Reset()

+ func (x *GatherDimensionNumbers) String() string

+ type GlobalDataHandle struct

+ Handle int64

+ func (*GlobalDataHandle) Descriptor() ([]byte, []int)

+ func (*GlobalDataHandle) ProtoMessage()

+ func (x *GlobalDataHandle) GetHandle() int64

+ func (x *GlobalDataHandle) ProtoReflect() protoreflect.Message

+ func (x *GlobalDataHandle) Reset()

+ func (x *GlobalDataHandle) String() string

+ type LayoutProto struct

+ ElementSizeInBits int64

+ Format Format

+ MemorySpace int64

+ MinorToMajor []int64

+ Tiles []*TileProto

+ func (*LayoutProto) Descriptor() ([]byte, []int)

+ func (*LayoutProto) ProtoMessage()

+ func (x *LayoutProto) GetElementSizeInBits() int64

+ func (x *LayoutProto) GetFormat() Format

+ func (x *LayoutProto) GetMemorySpace() int64

+ func (x *LayoutProto) GetMinorToMajor() []int64

+ func (x *LayoutProto) GetTiles() []*TileProto

+ func (x *LayoutProto) ProtoReflect() protoreflect.Message

+ func (x *LayoutProto) Reset()

+ func (x *LayoutProto) String() string

+ type LiteralProto struct

+ Bf16S []byte

+ C128S []float64

+ C64S []float32

+ F16S []byte

+ F32S []float32

+ F64S []float64

+ Preds []bool

+ S16S []byte

+ S32S []int32

+ S64S []int64

+ S8S []byte

+ Shape *ShapeProto

+ SparseIndices []int64

+ TupleLiterals []*LiteralProto

+ U16S []byte

+ U32S []uint32

+ U64S []uint64

+ U8S []byte

+ func (*LiteralProto) Descriptor() ([]byte, []int)

+ func (*LiteralProto) ProtoMessage()

+ func (x *LiteralProto) GetBf16S() []byte

+ func (x *LiteralProto) GetC128S() []float64

+ func (x *LiteralProto) GetC64S() []float32

+ func (x *LiteralProto) GetF16S() []byte

+ func (x *LiteralProto) GetF32S() []float32

+ func (x *LiteralProto) GetF64S() []float64

+ func (x *LiteralProto) GetPreds() []bool

+ func (x *LiteralProto) GetS16S() []byte

+ func (x *LiteralProto) GetS32S() []int32

+ func (x *LiteralProto) GetS64S() []int64

+ func (x *LiteralProto) GetS8S() []byte

+ func (x *LiteralProto) GetShape() *ShapeProto

+ func (x *LiteralProto) GetSparseIndices() []int64

+ func (x *LiteralProto) GetTupleLiterals() []*LiteralProto

+ func (x *LiteralProto) GetU16S() []byte

+ func (x *LiteralProto) GetU32S() []uint32

+ func (x *LiteralProto) GetU64S() []uint64

+ func (x *LiteralProto) GetU8S() []byte

+ func (x *LiteralProto) ProtoReflect() protoreflect.Message

+ func (x *LiteralProto) Reset()

+ func (x *LiteralProto) String() string

+ type OpMetadata struct

+ CreationPassId int64

+ LogicalCreationPassId int64

+ OpName string

+ OpType string

+ ProfileInfo *OpMetadata_ProfileInfo

+ ProfileType []ProfileType

+ ReplacedOp string

+ SizeOfGeneratedCodeInBytes int64

+ SizeOfMemoryWorkingSetInBytes int64

+ SourceFile string

+ SourceLine int32

+ func (*OpMetadata) Descriptor() ([]byte, []int)

+ func (*OpMetadata) ProtoMessage()

+ func (x *OpMetadata) GetCreationPassId() int64

+ func (x *OpMetadata) GetLogicalCreationPassId() int64

+ func (x *OpMetadata) GetOpName() string

+ func (x *OpMetadata) GetOpType() string

+ func (x *OpMetadata) GetProfileInfo() *OpMetadata_ProfileInfo

+ func (x *OpMetadata) GetProfileType() []ProfileType

+ func (x *OpMetadata) GetReplacedOp() string

+ func (x *OpMetadata) GetSizeOfGeneratedCodeInBytes() int64

+ func (x *OpMetadata) GetSizeOfMemoryWorkingSetInBytes() int64

+ func (x *OpMetadata) GetSourceFile() string

+ func (x *OpMetadata) GetSourceLine() int32

+ func (x *OpMetadata) ProtoReflect() protoreflect.Message

+ func (x *OpMetadata) Reset()

+ func (x *OpMetadata) String() string

+ type OpMetadata_ProfileInfo struct

+ CompilationEvent CompilationEvent

+ ProfileSource ProfileSource

+ ProfileType []ProfileType

+ RelativeSpeedup float64

+ func (*OpMetadata_ProfileInfo) Descriptor() ([]byte, []int)

+ func (*OpMetadata_ProfileInfo) ProtoMessage()

+ func (x *OpMetadata_ProfileInfo) GetCompilationEvent() CompilationEvent

+ func (x *OpMetadata_ProfileInfo) GetProfileSource() ProfileSource

+ func (x *OpMetadata_ProfileInfo) GetProfileType() []ProfileType

+ func (x *OpMetadata_ProfileInfo) GetRelativeSpeedup() float64

+ func (x *OpMetadata_ProfileInfo) ProtoReflect() protoreflect.Message

+ func (x *OpMetadata_ProfileInfo) Reset()

+ func (x *OpMetadata_ProfileInfo) String() string

+ type OpSharding struct

+ LastTileDims []OpSharding_Type

+ Metadata []*OpMetadata

+ ReplicateOnLastTileDim bool

+ TileAssignmentDevices []int64

+ TileAssignmentDimensions []int64

+ TileShape *ShapeProto

+ TupleShardings []*OpSharding

+ Type OpSharding_Type

+ func (*OpSharding) Descriptor() ([]byte, []int)

+ func (*OpSharding) ProtoMessage()

+ func (x *OpSharding) GetLastTileDims() []OpSharding_Type

+ func (x *OpSharding) GetMetadata() []*OpMetadata

+ func (x *OpSharding) GetReplicateOnLastTileDim() bool

+ func (x *OpSharding) GetTileAssignmentDevices() []int64

+ func (x *OpSharding) GetTileAssignmentDimensions() []int64

+ func (x *OpSharding) GetTileShape() *ShapeProto

+ func (x *OpSharding) GetTupleShardings() []*OpSharding

+ func (x *OpSharding) GetType() OpSharding_Type

+ func (x *OpSharding) ProtoReflect() protoreflect.Message

+ func (x *OpSharding) Reset()

+ func (x *OpSharding) String() string

+ type OpSharding_Type int32

+ const OpSharding_MANUAL

+ const OpSharding_MAXIMAL

+ const OpSharding_OTHER

+ const OpSharding_REPLICATED

+ const OpSharding_TUPLE

+ func (OpSharding_Type) Descriptor() protoreflect.EnumDescriptor

+ func (OpSharding_Type) EnumDescriptor() ([]byte, []int)

+ func (OpSharding_Type) Type() protoreflect.EnumType

+ func (x OpSharding_Type) Enum() *OpSharding_Type

+ func (x OpSharding_Type) Number() protoreflect.EnumNumber

+ func (x OpSharding_Type) String() string

+ type PaddingConfig struct

+ Dimensions []*PaddingConfig_PaddingConfigDimension

+ func (*PaddingConfig) Descriptor() ([]byte, []int)

+ func (*PaddingConfig) ProtoMessage()

+ func (x *PaddingConfig) GetDimensions() []*PaddingConfig_PaddingConfigDimension

+ func (x *PaddingConfig) ProtoReflect() protoreflect.Message

+ func (x *PaddingConfig) Reset()

+ func (x *PaddingConfig) String() string

+ type PaddingConfig_PaddingConfigDimension struct

+ EdgePaddingHigh int64

+ EdgePaddingLow int64

+ InteriorPadding int64

+ func (*PaddingConfig_PaddingConfigDimension) Descriptor() ([]byte, []int)

+ func (*PaddingConfig_PaddingConfigDimension) ProtoMessage()

+ func (x *PaddingConfig_PaddingConfigDimension) GetEdgePaddingHigh() int64

+ func (x *PaddingConfig_PaddingConfigDimension) GetEdgePaddingLow() int64

+ func (x *PaddingConfig_PaddingConfigDimension) GetInteriorPadding() int64

+ func (x *PaddingConfig_PaddingConfigDimension) ProtoReflect() protoreflect.Message

+ func (x *PaddingConfig_PaddingConfigDimension) Reset()

+ func (x *PaddingConfig_PaddingConfigDimension) String() string

+ type PaddingType int32

+ const PaddingType_PADDING_INVALID

+ const PaddingType_PADDING_SAME

+ const PaddingType_PADDING_VALID

+ func (PaddingType) Descriptor() protoreflect.EnumDescriptor

+ func (PaddingType) EnumDescriptor() ([]byte, []int)

+ func (PaddingType) Type() protoreflect.EnumType

+ func (x PaddingType) Enum() *PaddingType

+ func (x PaddingType) Number() protoreflect.EnumNumber

+ func (x PaddingType) String() string

+ type ParameterReplication struct

+ ReplicatedAtLeafBuffers []bool

+ func (*ParameterReplication) Descriptor() ([]byte, []int)

+ func (*ParameterReplication) ProtoMessage()

+ func (x *ParameterReplication) GetReplicatedAtLeafBuffers() []bool

+ func (x *ParameterReplication) ProtoReflect() protoreflect.Message

+ func (x *ParameterReplication) Reset()

+ func (x *ParameterReplication) String() string

+ type PrecisionConfig struct

+ OperandPrecision []PrecisionConfig_Precision

+ func (*PrecisionConfig) Descriptor() ([]byte, []int)

+ func (*PrecisionConfig) ProtoMessage()

+ func (x *PrecisionConfig) GetOperandPrecision() []PrecisionConfig_Precision

+ func (x *PrecisionConfig) ProtoReflect() protoreflect.Message

+ func (x *PrecisionConfig) Reset()

+ func (x *PrecisionConfig) String() string

+ type PrecisionConfig_Precision int32

+ const PrecisionConfig_DEFAULT

+ const PrecisionConfig_HIGH

+ const PrecisionConfig_HIGHEST

+ func (PrecisionConfig_Precision) Descriptor() protoreflect.EnumDescriptor

+ func (PrecisionConfig_Precision) EnumDescriptor() ([]byte, []int)

+ func (PrecisionConfig_Precision) Type() protoreflect.EnumType

+ func (x PrecisionConfig_Precision) Enum() *PrecisionConfig_Precision

+ func (x PrecisionConfig_Precision) Number() protoreflect.EnumNumber

+ func (x PrecisionConfig_Precision) String() string

+ type PrimitiveType int32

+ const PrimitiveType_BF16

+ const PrimitiveType_C128

+ const PrimitiveType_C64

+ const PrimitiveType_F16

+ const PrimitiveType_F32

+ const PrimitiveType_F64

+ const PrimitiveType_OPAQUE_TYPE

+ const PrimitiveType_PRED

+ const PrimitiveType_PRIMITIVE_TYPE_INVALID

+ const PrimitiveType_S16

+ const PrimitiveType_S32

+ const PrimitiveType_S64

+ const PrimitiveType_S8

+ const PrimitiveType_TOKEN

+ const PrimitiveType_TUPLE

+ const PrimitiveType_U16

+ const PrimitiveType_U32

+ const PrimitiveType_U64

+ const PrimitiveType_U8

+ func (PrimitiveType) Descriptor() protoreflect.EnumDescriptor

+ func (PrimitiveType) EnumDescriptor() ([]byte, []int)

+ func (PrimitiveType) Type() protoreflect.EnumType

+ func (x PrimitiveType) Enum() *PrimitiveType

+ func (x PrimitiveType) Number() protoreflect.EnumNumber

+ func (x PrimitiveType) String() string

+ type ProfileSource int32

+ const ProfileSource_PROFILE_SOURCE_EMBEDDED

+ const ProfileSource_PROFILE_SOURCE_REMOTE

+ const ProfileSource_PROFILE_SOURCE_UNKNOWN_SOURCE

+ func (ProfileSource) Descriptor() protoreflect.EnumDescriptor

+ func (ProfileSource) EnumDescriptor() ([]byte, []int)

+ func (ProfileSource) Type() protoreflect.EnumType

+ func (x ProfileSource) Enum() *ProfileSource

+ func (x ProfileSource) Number() protoreflect.EnumNumber

+ func (x ProfileSource) String() string

+ type ProfileType int32

+ const ProfileType_FLAG

+ const ProfileType_INTEGER

+ const ProfileType_INVALID

+ const ProfileType_WINDOW

+ func (ProfileType) Descriptor() protoreflect.EnumDescriptor

+ func (ProfileType) EnumDescriptor() ([]byte, []int)

+ func (ProfileType) Type() protoreflect.EnumType

+ func (x ProfileType) Enum() *ProfileType

+ func (x ProfileType) Number() protoreflect.EnumNumber

+ func (x ProfileType) String() string

+ type ProgramShapeProto struct

+ ParameterNames []string

+ Parameters []*ShapeProto

+ Result *ShapeProto

+ func (*ProgramShapeProto) Descriptor() ([]byte, []int)

+ func (*ProgramShapeProto) ProtoMessage()

+ func (x *ProgramShapeProto) GetParameterNames() []string

+ func (x *ProgramShapeProto) GetParameters() []*ShapeProto

+ func (x *ProgramShapeProto) GetResult() *ShapeProto

+ func (x *ProgramShapeProto) ProtoReflect() protoreflect.Message

+ func (x *ProgramShapeProto) Reset()

+ func (x *ProgramShapeProto) String() string

+ type RandomAlgorithm int32

+ const RandomAlgorithm_RNG_DEFAULT

+ const RandomAlgorithm_RNG_PHILOX

+ const RandomAlgorithm_RNG_THREE_FRY

+ func (RandomAlgorithm) Descriptor() protoreflect.EnumDescriptor

+ func (RandomAlgorithm) EnumDescriptor() ([]byte, []int)

+ func (RandomAlgorithm) Type() protoreflect.EnumType

+ func (x RandomAlgorithm) Enum() *RandomAlgorithm

+ func (x RandomAlgorithm) Number() protoreflect.EnumNumber

+ func (x RandomAlgorithm) String() string

+ type RandomDistribution int32

+ const RandomDistribution_RNG_INVALID

+ const RandomDistribution_RNG_NORMAL

+ const RandomDistribution_RNG_UNIFORM

+ func (RandomDistribution) Descriptor() protoreflect.EnumDescriptor

+ func (RandomDistribution) EnumDescriptor() ([]byte, []int)

+ func (RandomDistribution) Type() protoreflect.EnumType

+ func (x RandomDistribution) Enum() *RandomDistribution

+ func (x RandomDistribution) Number() protoreflect.EnumNumber

+ func (x RandomDistribution) String() string

+ type ReplicaGroup struct

+ ReplicaIds []int64

+ func (*ReplicaGroup) Descriptor() ([]byte, []int)

+ func (*ReplicaGroup) ProtoMessage()

+ func (x *ReplicaGroup) GetReplicaIds() []int64

+ func (x *ReplicaGroup) ProtoReflect() protoreflect.Message

+ func (x *ReplicaGroup) Reset()

+ func (x *ReplicaGroup) String() string

+ type ScatterDimensionNumbers struct

+ IndexVectorDim int64

+ InsertedWindowDims []int64

+ ScatterDimsToOperandDims []int64

+ UpdateWindowDims []int64

+ func (*ScatterDimensionNumbers) Descriptor() ([]byte, []int)

+ func (*ScatterDimensionNumbers) ProtoMessage()

+ func (x *ScatterDimensionNumbers) GetIndexVectorDim() int64

+ func (x *ScatterDimensionNumbers) GetInsertedWindowDims() []int64

+ func (x *ScatterDimensionNumbers) GetScatterDimsToOperandDims() []int64

+ func (x *ScatterDimensionNumbers) GetUpdateWindowDims() []int64

+ func (x *ScatterDimensionNumbers) ProtoReflect() protoreflect.Message

+ func (x *ScatterDimensionNumbers) Reset()

+ func (x *ScatterDimensionNumbers) String() string

+ type ShapeProto struct

+ Dimensions []int64

+ ElementType PrimitiveType

+ IsDynamicDimension []bool

+ Layout *LayoutProto

+ TupleShapes []*ShapeProto

+ func (*ShapeProto) Descriptor() ([]byte, []int)

+ func (*ShapeProto) ProtoMessage()

+ func (x *ShapeProto) GetDimensions() []int64

+ func (x *ShapeProto) GetElementType() PrimitiveType

+ func (x *ShapeProto) GetIsDynamicDimension() []bool

+ func (x *ShapeProto) GetLayout() *LayoutProto

+ func (x *ShapeProto) GetTupleShapes() []*ShapeProto

+ func (x *ShapeProto) ProtoReflect() protoreflect.Message

+ func (x *ShapeProto) Reset()

+ func (x *ShapeProto) String() string

+ type SourceTarget struct

+ Source int64

+ Target int64

+ func (*SourceTarget) Descriptor() ([]byte, []int)

+ func (*SourceTarget) ProtoMessage()

+ func (x *SourceTarget) GetSource() int64

+ func (x *SourceTarget) GetTarget() int64

+ func (x *SourceTarget) ProtoReflect() protoreflect.Message

+ func (x *SourceTarget) Reset()

+ func (x *SourceTarget) String() string

+ type TileProto struct

+ Dimensions []int64

+ func (*TileProto) Descriptor() ([]byte, []int)

+ func (*TileProto) ProtoMessage()

+ func (x *TileProto) GetDimensions() []int64

+ func (x *TileProto) ProtoReflect() protoreflect.Message

+ func (x *TileProto) Reset()

+ func (x *TileProto) String() string

+ type TriangularSolveOptions struct

+ LeftSide bool

+ Lower bool

+ TransposeA TriangularSolveOptions_Transpose

+ UnitDiagonal bool

+ func (*TriangularSolveOptions) Descriptor() ([]byte, []int)

+ func (*TriangularSolveOptions) ProtoMessage()

+ func (x *TriangularSolveOptions) GetLeftSide() bool

+ func (x *TriangularSolveOptions) GetLower() bool

+ func (x *TriangularSolveOptions) GetTransposeA() TriangularSolveOptions_Transpose

+ func (x *TriangularSolveOptions) GetUnitDiagonal() bool

+ func (x *TriangularSolveOptions) ProtoReflect() protoreflect.Message

+ func (x *TriangularSolveOptions) Reset()

+ func (x *TriangularSolveOptions) String() string

+ type TriangularSolveOptions_Transpose int32

+ const TriangularSolveOptions_ADJOINT

+ const TriangularSolveOptions_NO_TRANSPOSE

+ const TriangularSolveOptions_TRANSPOSE

+ const TriangularSolveOptions_TRANSPOSE_INVALID

+ func (TriangularSolveOptions_Transpose) Descriptor() protoreflect.EnumDescriptor

+ func (TriangularSolveOptions_Transpose) EnumDescriptor() ([]byte, []int)

+ func (TriangularSolveOptions_Transpose) Type() protoreflect.EnumType

+ func (x TriangularSolveOptions_Transpose) Enum() *TriangularSolveOptions_Transpose

+ func (x TriangularSolveOptions_Transpose) Number() protoreflect.EnumNumber

+ func (x TriangularSolveOptions_Transpose) String() string

+ type WhileLoopBackendConfig struct

+ KnownTripCount *WhileLoopBackendConfig_KnownTripCount

+ func (*WhileLoopBackendConfig) Descriptor() ([]byte, []int)

+ func (*WhileLoopBackendConfig) ProtoMessage()

+ func (x *WhileLoopBackendConfig) GetKnownTripCount() *WhileLoopBackendConfig_KnownTripCount

+ func (x *WhileLoopBackendConfig) ProtoReflect() protoreflect.Message

+ func (x *WhileLoopBackendConfig) Reset()

+ func (x *WhileLoopBackendConfig) String() string

+ type WhileLoopBackendConfig_KnownTripCount struct

+ N int64

+ func (*WhileLoopBackendConfig_KnownTripCount) Descriptor() ([]byte, []int)

+ func (*WhileLoopBackendConfig_KnownTripCount) ProtoMessage()

+ func (x *WhileLoopBackendConfig_KnownTripCount) GetN() int64

+ func (x *WhileLoopBackendConfig_KnownTripCount) ProtoReflect() protoreflect.Message

+ func (x *WhileLoopBackendConfig_KnownTripCount) Reset()

+ func (x *WhileLoopBackendConfig_KnownTripCount) String() string

+ type Window struct

+ Dimensions []*WindowDimension

+ func (*Window) Descriptor() ([]byte, []int)

+ func (*Window) ProtoMessage()

+ func (x *Window) GetDimensions() []*WindowDimension

+ func (x *Window) ProtoReflect() protoreflect.Message

+ func (x *Window) Reset()

+ func (x *Window) String() string

+ type WindowDimension struct

+ BaseDilation int64

+ PaddingHigh int64

+ PaddingLow int64

+ Size int64

+ Stride int64

+ WindowDilation int64

+ WindowReversal bool

+ func (*WindowDimension) Descriptor() ([]byte, []int)

+ func (*WindowDimension) ProtoMessage()

+ func (x *WindowDimension) GetBaseDilation() int64

+ func (x *WindowDimension) GetPaddingHigh() int64

+ func (x *WindowDimension) GetPaddingLow() int64

+ func (x *WindowDimension) GetSize() int64

+ func (x *WindowDimension) GetStride() int64

+ func (x *WindowDimension) GetWindowDilation() int64

+ func (x *WindowDimension) GetWindowReversal() bool

+ func (x *WindowDimension) ProtoReflect() protoreflect.Message

+ func (x *WindowDimension) Reset()

+ func (x *WindowDimension) String() string

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