Documentation ¶
Index ¶
- Variables
- type CompileRequest
- func (*CompileRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CompileRequest) GetComputation() *service.HloModuleProto
- func (x *CompileRequest) GetExecutionOptions() *ExecutionOptions
- func (x *CompileRequest) GetInputShapeWithLayout() []*data.ShapeProto
- func (*CompileRequest) ProtoMessage()
- func (x *CompileRequest) ProtoReflect() protoreflect.Message
- func (x *CompileRequest) Reset()
- func (x *CompileRequest) String() string
- type CompileResponse
- type ComputationGraphStatsRequest
- func (*ComputationGraphStatsRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ComputationGraphStatsRequest) GetComputation() *service.HloModuleProto
- func (x *ComputationGraphStatsRequest) GetDebugOptions() *DebugOptions
- func (*ComputationGraphStatsRequest) ProtoMessage()
- func (x *ComputationGraphStatsRequest) ProtoReflect() protoreflect.Message
- func (x *ComputationGraphStatsRequest) Reset()
- func (x *ComputationGraphStatsRequest) String() string
- type ComputationStatsResponse
- func (*ComputationStatsResponse) Descriptor() ([]byte, []int)deprecated
- func (x *ComputationStatsResponse) GetStats() *data.ComputationStats
- func (*ComputationStatsResponse) ProtoMessage()
- func (x *ComputationStatsResponse) ProtoReflect() protoreflect.Message
- func (x *ComputationStatsResponse) Reset()
- func (x *ComputationStatsResponse) String() string
- type ComputeConstantGraphRequest
- func (*ComputeConstantGraphRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ComputeConstantGraphRequest) GetComputation() *service.HloModuleProto
- func (x *ComputeConstantGraphRequest) GetOutputLayout() *data.LayoutProto
- func (*ComputeConstantGraphRequest) ProtoMessage()
- func (x *ComputeConstantGraphRequest) ProtoReflect() protoreflect.Message
- func (x *ComputeConstantGraphRequest) Reset()
- func (x *ComputeConstantGraphRequest) String() string
- type ComputeConstantResponse
- func (*ComputeConstantResponse) Descriptor() ([]byte, []int)deprecated
- func (x *ComputeConstantResponse) GetLiteral() *data.LiteralProto
- func (*ComputeConstantResponse) ProtoMessage()
- func (x *ComputeConstantResponse) ProtoReflect() protoreflect.Message
- func (x *ComputeConstantResponse) Reset()
- func (x *ComputeConstantResponse) String() string
- type CreateChannelHandleRequest
- func (*CreateChannelHandleRequest) Descriptor() ([]byte, []int)deprecated
- func (x *CreateChannelHandleRequest) GetChannelType() data.ChannelHandle_ChannelType
- func (*CreateChannelHandleRequest) ProtoMessage()
- func (x *CreateChannelHandleRequest) ProtoReflect() protoreflect.Message
- func (x *CreateChannelHandleRequest) Reset()
- func (x *CreateChannelHandleRequest) String() string
- type CreateChannelHandleResponse
- func (*CreateChannelHandleResponse) Descriptor() ([]byte, []int)deprecated
- func (x *CreateChannelHandleResponse) GetChannel() *data.ChannelHandle
- func (*CreateChannelHandleResponse) ProtoMessage()
- func (x *CreateChannelHandleResponse) ProtoReflect() protoreflect.Message
- func (x *CreateChannelHandleResponse) Reset()
- func (x *CreateChannelHandleResponse) String() string
- type DebugOptions
- func (*DebugOptions) Descriptor() ([]byte, []int)deprecated
- func (x *DebugOptions) GetXlaAllowExcessPrecision() bool
- func (x *DebugOptions) GetXlaAllowScalarIndexDynamicOps() bool
- func (x *DebugOptions) GetXlaBackendExtraOptions() map[string]string
- func (x *DebugOptions) GetXlaBackendOptimizationLevel() int32
- func (x *DebugOptions) GetXlaCpuEnableFastMath() bool
- func (x *DebugOptions) GetXlaCpuEnableFastMinMax() bool
- func (x *DebugOptions) GetXlaCpuEnableMlirLowering() bool
- func (x *DebugOptions) GetXlaCpuEnableXprofTraceme() bool
- func (x *DebugOptions) GetXlaCpuFastMathHonorDivision() bool
- func (x *DebugOptions) GetXlaCpuFastMathHonorFunctions() bool
- func (x *DebugOptions) GetXlaCpuFastMathHonorInfs() bool
- func (x *DebugOptions) GetXlaCpuFastMathHonorNans() bool
- func (x *DebugOptions) GetXlaCpuMultiThreadEigen() bool
- func (x *DebugOptions) GetXlaCpuStrictDotConvMath() bool
- func (x *DebugOptions) GetXlaCpuUseAcl() bool
- func (x *DebugOptions) GetXlaCpuUseJitrt() bool
- func (x *DebugOptions) GetXlaCpuUseMklDnn() bool
- func (x *DebugOptions) GetXlaDetailedLoggingAndDumping() bool
- func (x *DebugOptions) GetXlaDisableAllHloPasses() bool
- func (x *DebugOptions) GetXlaDisableHloPasses() []string
- func (x *DebugOptions) GetXlaDumpCompressProtos() bool
- func (x *DebugOptions) GetXlaDumpDisableMetadata() bool
- func (x *DebugOptions) GetXlaDumpFusionVisualization() bool
- func (x *DebugOptions) GetXlaDumpHloAsDot() bool
- func (x *DebugOptions) GetXlaDumpHloAsHtml() bool
- func (x *DebugOptions) GetXlaDumpHloAsLongText() bool
- func (x *DebugOptions) GetXlaDumpHloAsProto() bool
- func (x *DebugOptions) GetXlaDumpHloAsText() bool
- func (x *DebugOptions) GetXlaDumpHloAsUrl() bool
- func (x *DebugOptions) GetXlaDumpHloModuleRe() string
- func (x *DebugOptions) GetXlaDumpHloPassRe() string
- func (x *DebugOptions) GetXlaDumpHloPipelineRe() string
- func (x *DebugOptions) GetXlaDumpHloSnapshots() bool
- func (x *DebugOptions) GetXlaDumpIncludeTimestamp() bool
- func (x *DebugOptions) GetXlaDumpMaxHloModules() int32
- func (x *DebugOptions) GetXlaDumpModuleMetadata() bool
- func (x *DebugOptions) GetXlaDumpTo() string
- func (x *DebugOptions) GetXlaEliminateHloImplicitBroadcast() bool
- func (x *DebugOptions) GetXlaEmbedIrInExecutable() bool
- func (x *DebugOptions) GetXlaEnableHloPassesOnly() []string
- func (x *DebugOptions) GetXlaForceHostPlatformDeviceCount() int32
- func (x *DebugOptions) GetXlaGpuAlgorithmDenylistPath() string
- func (x *DebugOptions) GetXlaGpuAllReduceBlueconnectNumDevicesPerHost() int32
- func (x *DebugOptions) GetXlaGpuAllReduceCombineThresholdBytes() int64
- func (x *DebugOptions) GetXlaGpuAllReduceContiguous() bool
- func (x *DebugOptions) GetXlaGpuAsmExtraFlags() string
- func (x *DebugOptions) GetXlaGpuAutotuneLevel() int32
- func (x *DebugOptions) GetXlaGpuBefExecutable() bool
- func (x *DebugOptions) GetXlaGpuBefThunk() bool
- func (x *DebugOptions) GetXlaGpuCrashOnVerificationFailures() bool
- func (x *DebugOptions) GetXlaGpuCudaDataDir() string
- func (x *DebugOptions) GetXlaGpuDeterministicOps() bool
- func (x *DebugOptions) GetXlaGpuDisableGpuasmOptimizations() bool
- func (x *DebugOptions) GetXlaGpuDisableMultiStreaming() bool
- func (x *DebugOptions) GetXlaGpuDumpLlvmir() bool
- func (x *DebugOptions) GetXlaGpuEnableAsyncAllReduce() bool
- func (x *DebugOptions) GetXlaGpuEnableCublaslt() bool
- func (x *DebugOptions) GetXlaGpuEnableCudnnFrontend() bool
- func (x *DebugOptions) GetXlaGpuEnableFastMinMax() bool
- func (x *DebugOptions) GetXlaGpuEnableMlirLowering() bool
- func (x *DebugOptions) GetXlaGpuEnableSharedConstants() bool
- func (x *DebugOptions) GetXlaGpuForceCompilationParallelism() int32
- func (x *DebugOptions) GetXlaGpuForceConvNchw() bool
- func (x *DebugOptions) GetXlaGpuForceConvNhwc() bool
- func (x *DebugOptions) GetXlaGpuFtz() bool
- func (x *DebugOptions) GetXlaGpuJitrtExecutable() bool
- func (x *DebugOptions) GetXlaGpuLlvmIrFile() []string
- func (x *DebugOptions) GetXlaGpuMaxKernelUnrollFactor() int32
- func (x *DebugOptions) GetXlaGpuNcclTerminationTimeoutSeconds() int64
- func (x *DebugOptions) GetXlaGpuNormalizeLayouts() bool
- func (x *DebugOptions) GetXlaGpuPtxFile() []string
- func (x *DebugOptions) GetXlaGpuRedzoneScratchMaxMegabytes() int64
- func (x *DebugOptions) GetXlaGpuShapeChecks() DebugOptions_ShapeChecks
- func (x *DebugOptions) GetXlaGpuSimplifyAllFpConversions() bool
- func (x *DebugOptions) GetXlaGpuSimplifyGathers() bool
- func (x *DebugOptions) GetXlaGpuSimplifyScatters() bool
- func (x *DebugOptions) GetXlaGpuStrictConvAlgorithmPicker() bool
- func (x *DebugOptions) GetXlaGpuUnsafeFallbackToDriverOnPtxasNotFound() bool
- func (x *DebugOptions) GetXlaGpuUseRandomStreams() bool
- func (x *DebugOptions) GetXlaHloEvaluatorUseFastPath() bool
- func (x *DebugOptions) GetXlaHloGraphAddresses() bool
- func (x *DebugOptions) GetXlaHloGraphShardingColor() bool
- func (x *DebugOptions) GetXlaHloProfile() bool
- func (x *DebugOptions) GetXlaLlvmDisableExpensivePasses() bool
- func (x *DebugOptions) GetXlaLlvmEnableAliasScopeMetadata() bool
- func (x *DebugOptions) GetXlaLlvmEnableInvariantLoadMetadata() bool
- func (x *DebugOptions) GetXlaLlvmEnableNoaliasMetadata() bool
- func (x *DebugOptions) GetXlaMultiheapSizeConstraintPerHeap() int32
- func (x *DebugOptions) GetXlaStepMarkerLocation() DebugOptions_StepMarkerLocation
- func (x *DebugOptions) GetXlaTestAllInputLayouts() bool
- func (x *DebugOptions) GetXlaTestAllOutputLayouts() bool
- func (x *DebugOptions) GetXlaTpuDetectInf() bool
- func (x *DebugOptions) GetXlaTpuDetectNan() bool
- func (*DebugOptions) ProtoMessage()
- func (x *DebugOptions) ProtoReflect() protoreflect.Message
- func (x *DebugOptions) Reset()
- func (x *DebugOptions) String() string
- type DebugOptions_ShapeChecks
- func (DebugOptions_ShapeChecks) Descriptor() protoreflect.EnumDescriptor
- func (x DebugOptions_ShapeChecks) Enum() *DebugOptions_ShapeChecks
- func (DebugOptions_ShapeChecks) EnumDescriptor() ([]byte, []int)deprecated
- func (x DebugOptions_ShapeChecks) Number() protoreflect.EnumNumber
- func (x DebugOptions_ShapeChecks) String() string
- func (DebugOptions_ShapeChecks) Type() protoreflect.EnumType
- type DebugOptions_StepMarkerLocation
- func (DebugOptions_StepMarkerLocation) Descriptor() protoreflect.EnumDescriptor
- func (x DebugOptions_StepMarkerLocation) Enum() *DebugOptions_StepMarkerLocation
- func (DebugOptions_StepMarkerLocation) EnumDescriptor() ([]byte, []int)deprecated
- func (x DebugOptions_StepMarkerLocation) Number() protoreflect.EnumNumber
- func (x DebugOptions_StepMarkerLocation) String() string
- func (DebugOptions_StepMarkerLocation) Type() protoreflect.EnumType
- type DeconstructTupleRequest
- func (*DeconstructTupleRequest) Descriptor() ([]byte, []int)deprecated
- func (x *DeconstructTupleRequest) GetTupleHandle() *data.GlobalDataHandle
- func (*DeconstructTupleRequest) ProtoMessage()
- func (x *DeconstructTupleRequest) ProtoReflect() protoreflect.Message
- func (x *DeconstructTupleRequest) Reset()
- func (x *DeconstructTupleRequest) String() string
- type DeconstructTupleResponse
- func (*DeconstructTupleResponse) Descriptor() ([]byte, []int)deprecated
- func (x *DeconstructTupleResponse) GetElementHandles() []*data.GlobalDataHandle
- func (*DeconstructTupleResponse) ProtoMessage()
- func (x *DeconstructTupleResponse) ProtoReflect() protoreflect.Message
- func (x *DeconstructTupleResponse) Reset()
- func (x *DeconstructTupleResponse) String() string
- type ExecuteGraphParallelRequest
- func (*ExecuteGraphParallelRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ExecuteGraphParallelRequest) GetRequests() []*ExecuteGraphRequest
- func (*ExecuteGraphParallelRequest) ProtoMessage()
- func (x *ExecuteGraphParallelRequest) ProtoReflect() protoreflect.Message
- func (x *ExecuteGraphParallelRequest) Reset()
- func (x *ExecuteGraphParallelRequest) String() string
- type ExecuteGraphRequest
- func (*ExecuteGraphRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ExecuteGraphRequest) GetArguments() []*data.GlobalDataHandle
- func (x *ExecuteGraphRequest) GetComputation() *service.HloModuleProto
- func (x *ExecuteGraphRequest) GetExecutionOptions() *ExecutionOptions
- func (*ExecuteGraphRequest) ProtoMessage()
- func (x *ExecuteGraphRequest) ProtoReflect() protoreflect.Message
- func (x *ExecuteGraphRequest) Reset()
- func (x *ExecuteGraphRequest) String() string
- type ExecuteParallelResponse
- func (*ExecuteParallelResponse) Descriptor() ([]byte, []int)deprecated
- func (x *ExecuteParallelResponse) GetResponses() []*ExecuteResponse
- func (*ExecuteParallelResponse) ProtoMessage()
- func (x *ExecuteParallelResponse) ProtoReflect() protoreflect.Message
- func (x *ExecuteParallelResponse) Reset()
- func (x *ExecuteParallelResponse) String() string
- type ExecuteRequest
- func (*ExecuteRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ExecuteRequest) GetArguments() []*data.GlobalDataHandle
- func (x *ExecuteRequest) GetHandle() *data.ExecutionHandle
- func (*ExecuteRequest) ProtoMessage()
- func (x *ExecuteRequest) ProtoReflect() protoreflect.Message
- func (x *ExecuteRequest) Reset()
- func (x *ExecuteRequest) String() string
- type ExecuteResponse
- func (*ExecuteResponse) Descriptor() ([]byte, []int)deprecated
- func (x *ExecuteResponse) GetOutput() *data.GlobalDataHandle
- func (x *ExecuteResponse) GetProfile() *data.ExecutionProfile
- func (*ExecuteResponse) ProtoMessage()
- func (x *ExecuteResponse) ProtoReflect() protoreflect.Message
- func (x *ExecuteResponse) Reset()
- func (x *ExecuteResponse) String() string
- type ExecutionOptions
- func (*ExecutionOptions) Descriptor() ([]byte, []int)deprecated
- func (x *ExecutionOptions) GetAliasPassthroughParams() bool
- func (x *ExecutionOptions) GetAllowSpmdShardingPropagationToOutput() bool
- func (x *ExecutionOptions) GetAutoSpmdPartitioningMeshIds() []int64
- func (x *ExecutionOptions) GetAutoSpmdPartitioningMeshShape() []int64
- func (x *ExecutionOptions) GetDebugOptions() *DebugOptions
- func (x *ExecutionOptions) GetDeduplicateHlo() bool
- func (x *ExecutionOptions) GetDeviceAssignment() *data.DeviceAssignmentProto
- func (x *ExecutionOptions) GetDeviceHandles() []*data.DeviceHandle
- func (x *ExecutionOptions) GetLaunchId() int32
- func (x *ExecutionOptions) GetNumPartitions() int32
- func (x *ExecutionOptions) GetNumReplicas() int32
- func (x *ExecutionOptions) GetSeed() uint64
- func (x *ExecutionOptions) GetShapeWithOutputLayout() *data.ShapeProto
- func (x *ExecutionOptions) GetUseAutoSpmdPartitioning() bool
- func (x *ExecutionOptions) GetUseSpmdPartitioning() bool
- func (*ExecutionOptions) ProtoMessage()
- func (x *ExecutionOptions) ProtoReflect() protoreflect.Message
- func (x *ExecutionOptions) Reset()
- func (x *ExecutionOptions) String() string
- type GetDeviceHandlesRequest
- func (*GetDeviceHandlesRequest) Descriptor() ([]byte, []int)deprecated
- func (x *GetDeviceHandlesRequest) GetDeviceCount() int64
- func (*GetDeviceHandlesRequest) ProtoMessage()
- func (x *GetDeviceHandlesRequest) ProtoReflect() protoreflect.Message
- func (x *GetDeviceHandlesRequest) Reset()
- func (x *GetDeviceHandlesRequest) String() string
- type GetDeviceHandlesResponse
- func (*GetDeviceHandlesResponse) Descriptor() ([]byte, []int)deprecated
- func (x *GetDeviceHandlesResponse) GetDeviceHandles() []*data.DeviceHandle
- func (*GetDeviceHandlesResponse) ProtoMessage()
- func (x *GetDeviceHandlesResponse) ProtoReflect() protoreflect.Message
- func (x *GetDeviceHandlesResponse) Reset()
- func (x *GetDeviceHandlesResponse) String() string
- type GetShapeRequest
- type GetShapeResponse
- type LoadDataRequest
- func (*LoadDataRequest) Descriptor() ([]byte, []int)deprecated
- func (x *LoadDataRequest) GetColumnioField() string
- func (x *LoadDataRequest) GetColumnioTabletPath() string
- func (x *LoadDataRequest) GetElementShape() *data.ShapeProto
- func (x *LoadDataRequest) GetLimit() int64
- func (x *LoadDataRequest) GetOffset() int64
- func (x *LoadDataRequest) GetZip() bool
- func (*LoadDataRequest) ProtoMessage()
- func (x *LoadDataRequest) ProtoReflect() protoreflect.Message
- func (x *LoadDataRequest) Reset()
- func (x *LoadDataRequest) String() string
- type LoadDataResponse
- func (*LoadDataResponse) Descriptor() ([]byte, []int)deprecated
- func (x *LoadDataResponse) GetAvailableRows() int64
- func (x *LoadDataResponse) GetData() *data.GlobalDataHandle
- func (x *LoadDataResponse) GetDataShape() *data.ShapeProto
- func (x *LoadDataResponse) GetNanoseconds() int64
- func (x *LoadDataResponse) GetRowsLoaded() int64
- func (*LoadDataResponse) ProtoMessage()
- func (x *LoadDataResponse) ProtoReflect() protoreflect.Message
- func (x *LoadDataResponse) Reset()
- func (x *LoadDataResponse) String() string
- type ResetDeviceRequest
- func (*ResetDeviceRequest) Descriptor() ([]byte, []int)deprecated
- func (x *ResetDeviceRequest) GetDeviceHandle() *data.DeviceHandle
- func (*ResetDeviceRequest) ProtoMessage()
- func (x *ResetDeviceRequest) ProtoReflect() protoreflect.Message
- func (x *ResetDeviceRequest) Reset()
- func (x *ResetDeviceRequest) String() string
- type ResetDeviceResponse
- type TransferFromOutfeedRequest
- func (*TransferFromOutfeedRequest) Descriptor() ([]byte, []int)deprecated
- func (x *TransferFromOutfeedRequest) GetDeviceHandle() *data.DeviceHandle
- func (x *TransferFromOutfeedRequest) GetReplicaId() int64
- func (x *TransferFromOutfeedRequest) GetShapeWithLayout() *data.ShapeProto
- func (*TransferFromOutfeedRequest) ProtoMessage()
- func (x *TransferFromOutfeedRequest) ProtoReflect() protoreflect.Message
- func (x *TransferFromOutfeedRequest) Reset()
- func (x *TransferFromOutfeedRequest) String() string
- type TransferFromOutfeedResponse
- func (*TransferFromOutfeedResponse) Descriptor() ([]byte, []int)deprecated
- func (x *TransferFromOutfeedResponse) GetLiteral() *data.LiteralProto
- func (*TransferFromOutfeedResponse) ProtoMessage()
- func (x *TransferFromOutfeedResponse) ProtoReflect() protoreflect.Message
- func (x *TransferFromOutfeedResponse) Reset()
- func (x *TransferFromOutfeedResponse) String() string
- type TransferToClientRequest
- func (*TransferToClientRequest) Descriptor() ([]byte, []int)deprecated
- func (x *TransferToClientRequest) GetData() *data.GlobalDataHandle
- func (x *TransferToClientRequest) GetShapeWithLayout() *data.ShapeProto
- func (*TransferToClientRequest) ProtoMessage()
- func (x *TransferToClientRequest) ProtoReflect() protoreflect.Message
- func (x *TransferToClientRequest) Reset()
- func (x *TransferToClientRequest) String() string
- type TransferToClientResponse
- func (*TransferToClientResponse) Descriptor() ([]byte, []int)deprecated
- func (x *TransferToClientResponse) GetLiteral() *data.LiteralProto
- func (*TransferToClientResponse) ProtoMessage()
- func (x *TransferToClientResponse) ProtoReflect() protoreflect.Message
- func (x *TransferToClientResponse) Reset()
- func (x *TransferToClientResponse) String() string
- type TransferToInfeedRequest
- func (*TransferToInfeedRequest) Descriptor() ([]byte, []int)deprecated
- func (x *TransferToInfeedRequest) GetDeviceHandle() *data.DeviceHandle
- func (x *TransferToInfeedRequest) GetLiteral() *data.LiteralProto
- func (x *TransferToInfeedRequest) GetReplicaId() int64
- func (*TransferToInfeedRequest) ProtoMessage()
- func (x *TransferToInfeedRequest) ProtoReflect() protoreflect.Message
- func (x *TransferToInfeedRequest) Reset()
- func (x *TransferToInfeedRequest) String() string
- type TransferToInfeedResponse
- type TransferToServerRequest
- func (*TransferToServerRequest) Descriptor() ([]byte, []int)deprecated
- func (x *TransferToServerRequest) GetDeviceHandle() *data.DeviceHandle
- func (x *TransferToServerRequest) GetLiteral() *data.LiteralProto
- func (*TransferToServerRequest) ProtoMessage()
- func (x *TransferToServerRequest) ProtoReflect() protoreflect.Message
- func (x *TransferToServerRequest) Reset()
- func (x *TransferToServerRequest) String() string
- type TransferToServerResponse
- func (*TransferToServerResponse) Descriptor() ([]byte, []int)deprecated
- func (x *TransferToServerResponse) GetData() *data.GlobalDataHandle
- func (*TransferToServerResponse) ProtoMessage()
- func (x *TransferToServerResponse) ProtoReflect() protoreflect.Message
- func (x *TransferToServerResponse) Reset()
- func (x *TransferToServerResponse) String() string
- type UnpackRequest
- type UnpackResponse
- type UnregisterRequest
- func (*UnregisterRequest) Descriptor() ([]byte, []int)deprecated
- func (x *UnregisterRequest) GetData() []*data.GlobalDataHandle
- func (*UnregisterRequest) ProtoMessage()
- func (x *UnregisterRequest) ProtoReflect() protoreflect.Message
- func (x *UnregisterRequest) Reset()
- func (x *UnregisterRequest) String() string
- type UnregisterResponse
- type WaitForExecutionRequest
- func (*WaitForExecutionRequest) Descriptor() ([]byte, []int)deprecated
- func (x *WaitForExecutionRequest) GetExecution() *data.ExecutionHandle
- func (*WaitForExecutionRequest) ProtoMessage()
- func (x *WaitForExecutionRequest) ProtoReflect() protoreflect.Message
- func (x *WaitForExecutionRequest) Reset()
- func (x *WaitForExecutionRequest) String() string
- type WaitForExecutionResponse
- func (*WaitForExecutionResponse) Descriptor() ([]byte, []int)deprecated
- func (x *WaitForExecutionResponse) GetOutput() *data.GlobalDataHandle
- func (x *WaitForExecutionResponse) GetProfile() *data.ExecutionProfile
- func (*WaitForExecutionResponse) ProtoMessage()
- func (x *WaitForExecutionResponse) ProtoReflect() protoreflect.Message
- func (x *WaitForExecutionResponse) Reset()
- func (x *WaitForExecutionResponse) String() string
Constants ¶
This section is empty.
Variables ¶
var ( DebugOptions_ShapeChecks_name = map[int32]string{ 0: "IGNORE", 1: "RUNTIME", 2: "COMPILE_TIME", } DebugOptions_ShapeChecks_value = map[string]int32{ "IGNORE": 0, "RUNTIME": 1, "COMPILE_TIME": 2, } )
Enum value maps for DebugOptions_ShapeChecks.
var ( DebugOptions_StepMarkerLocation_name = map[int32]string{ 0: "STEP_MARK_AT_ENTRY", 1: "STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP", 3: "STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP", 2: "STEP_MARK_NONE", } DebugOptions_StepMarkerLocation_value = map[string]int32{ "STEP_MARK_AT_ENTRY": 0, "STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP": 1, "STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP": 3, "STEP_MARK_NONE": 2, } )
Enum value maps for DebugOptions_StepMarkerLocation.
var File_tensorflow_compiler_xla_xla_proto protoreflect.FileDescriptor
Functions ¶
This section is empty.
Types ¶
type CompileRequest ¶
type CompileRequest struct { // The graph to be compiled. Computation *service.HloModuleProto `protobuf:"bytes,1,opt,name=computation,proto3" json:"computation,omitempty"` // Options that affect how XLA compiles code to service this request. ExecutionOptions *ExecutionOptions `protobuf:"bytes,2,opt,name=execution_options,json=executionOptions,proto3" json:"execution_options,omitempty"` // The layouts of the input arguments. If not set, the default layout will be // used. Although the real arguments are not needed in compilation, the // layouts of the arguments can affect the compilation. InputShapeWithLayout []*data.ShapeProto `protobuf:"bytes,3,rep,name=input_shape_with_layout,json=inputShapeWithLayout,proto3" json:"input_shape_with_layout,omitempty"` // contains filtered or unexported fields }
func (*CompileRequest) Descriptor
deprecated
func (*CompileRequest) Descriptor() ([]byte, []int)
Deprecated: Use CompileRequest.ProtoReflect.Descriptor instead.
func (*CompileRequest) GetComputation ¶
func (x *CompileRequest) GetComputation() *service.HloModuleProto
func (*CompileRequest) GetExecutionOptions ¶
func (x *CompileRequest) GetExecutionOptions() *ExecutionOptions
func (*CompileRequest) GetInputShapeWithLayout ¶
func (x *CompileRequest) GetInputShapeWithLayout() []*data.ShapeProto
func (*CompileRequest) ProtoMessage ¶
func (*CompileRequest) ProtoMessage()
func (*CompileRequest) ProtoReflect ¶
func (x *CompileRequest) ProtoReflect() protoreflect.Message
func (*CompileRequest) Reset ¶
func (x *CompileRequest) Reset()
func (*CompileRequest) String ¶
func (x *CompileRequest) String() string
type CompileResponse ¶
type CompileResponse struct { // The handle to the executable. Handle *data.ExecutionHandle `protobuf:"bytes,1,opt,name=handle,proto3" json:"handle,omitempty"` // contains filtered or unexported fields }
func (*CompileResponse) Descriptor
deprecated
func (*CompileResponse) Descriptor() ([]byte, []int)
Deprecated: Use CompileResponse.ProtoReflect.Descriptor instead.
func (*CompileResponse) GetHandle ¶
func (x *CompileResponse) GetHandle() *data.ExecutionHandle
func (*CompileResponse) ProtoMessage ¶
func (*CompileResponse) ProtoMessage()
func (*CompileResponse) ProtoReflect ¶
func (x *CompileResponse) ProtoReflect() protoreflect.Message
func (*CompileResponse) Reset ¶
func (x *CompileResponse) Reset()
func (*CompileResponse) String ¶
func (x *CompileResponse) String() string
type ComputationGraphStatsRequest ¶
type ComputationGraphStatsRequest struct { Computation *service.HloModuleProto `protobuf:"bytes,1,opt,name=computation,proto3" json:"computation,omitempty"` DebugOptions *DebugOptions `protobuf:"bytes,2,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"` // contains filtered or unexported fields }
func (*ComputationGraphStatsRequest) Descriptor
deprecated
func (*ComputationGraphStatsRequest) Descriptor() ([]byte, []int)
Deprecated: Use ComputationGraphStatsRequest.ProtoReflect.Descriptor instead.
func (*ComputationGraphStatsRequest) GetComputation ¶
func (x *ComputationGraphStatsRequest) GetComputation() *service.HloModuleProto
func (*ComputationGraphStatsRequest) GetDebugOptions ¶
func (x *ComputationGraphStatsRequest) GetDebugOptions() *DebugOptions
func (*ComputationGraphStatsRequest) ProtoMessage ¶
func (*ComputationGraphStatsRequest) ProtoMessage()
func (*ComputationGraphStatsRequest) ProtoReflect ¶
func (x *ComputationGraphStatsRequest) ProtoReflect() protoreflect.Message
func (*ComputationGraphStatsRequest) Reset ¶
func (x *ComputationGraphStatsRequest) Reset()
func (*ComputationGraphStatsRequest) String ¶
func (x *ComputationGraphStatsRequest) String() string
type ComputationStatsResponse ¶
type ComputationStatsResponse struct { Stats *data.ComputationStats `protobuf:"bytes,1,opt,name=stats,proto3" json:"stats,omitempty"` // contains filtered or unexported fields }
func (*ComputationStatsResponse) Descriptor
deprecated
func (*ComputationStatsResponse) Descriptor() ([]byte, []int)
Deprecated: Use ComputationStatsResponse.ProtoReflect.Descriptor instead.
func (*ComputationStatsResponse) GetStats ¶
func (x *ComputationStatsResponse) GetStats() *data.ComputationStats
func (*ComputationStatsResponse) ProtoMessage ¶
func (*ComputationStatsResponse) ProtoMessage()
func (*ComputationStatsResponse) ProtoReflect ¶
func (x *ComputationStatsResponse) ProtoReflect() protoreflect.Message
func (*ComputationStatsResponse) Reset ¶
func (x *ComputationStatsResponse) Reset()
func (*ComputationStatsResponse) String ¶
func (x *ComputationStatsResponse) String() string
type ComputeConstantGraphRequest ¶
type ComputeConstantGraphRequest struct { Computation *service.HloModuleProto `protobuf:"bytes,1,opt,name=computation,proto3" json:"computation,omitempty"` OutputLayout *data.LayoutProto `protobuf:"bytes,2,opt,name=output_layout,json=outputLayout,proto3" json:"output_layout,omitempty"` // contains filtered or unexported fields }
func (*ComputeConstantGraphRequest) Descriptor
deprecated
func (*ComputeConstantGraphRequest) Descriptor() ([]byte, []int)
Deprecated: Use ComputeConstantGraphRequest.ProtoReflect.Descriptor instead.
func (*ComputeConstantGraphRequest) GetComputation ¶
func (x *ComputeConstantGraphRequest) GetComputation() *service.HloModuleProto
func (*ComputeConstantGraphRequest) GetOutputLayout ¶
func (x *ComputeConstantGraphRequest) GetOutputLayout() *data.LayoutProto
func (*ComputeConstantGraphRequest) ProtoMessage ¶
func (*ComputeConstantGraphRequest) ProtoMessage()
func (*ComputeConstantGraphRequest) ProtoReflect ¶
func (x *ComputeConstantGraphRequest) ProtoReflect() protoreflect.Message
func (*ComputeConstantGraphRequest) Reset ¶
func (x *ComputeConstantGraphRequest) Reset()
func (*ComputeConstantGraphRequest) String ¶
func (x *ComputeConstantGraphRequest) String() string
type ComputeConstantResponse ¶
type ComputeConstantResponse struct { // A LiteralProto is returned directly for this request. Literal *data.LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"` // contains filtered or unexported fields }
func (*ComputeConstantResponse) Descriptor
deprecated
func (*ComputeConstantResponse) Descriptor() ([]byte, []int)
Deprecated: Use ComputeConstantResponse.ProtoReflect.Descriptor instead.
func (*ComputeConstantResponse) GetLiteral ¶
func (x *ComputeConstantResponse) GetLiteral() *data.LiteralProto
func (*ComputeConstantResponse) ProtoMessage ¶
func (*ComputeConstantResponse) ProtoMessage()
func (*ComputeConstantResponse) ProtoReflect ¶
func (x *ComputeConstantResponse) ProtoReflect() protoreflect.Message
func (*ComputeConstantResponse) Reset ¶
func (x *ComputeConstantResponse) Reset()
func (*ComputeConstantResponse) String ¶
func (x *ComputeConstantResponse) String() string
type CreateChannelHandleRequest ¶
type CreateChannelHandleRequest struct { ChannelType data.ChannelHandle_ChannelType `` /* 130-byte string literal not displayed */ // contains filtered or unexported fields }
func (*CreateChannelHandleRequest) Descriptor
deprecated
func (*CreateChannelHandleRequest) Descriptor() ([]byte, []int)
Deprecated: Use CreateChannelHandleRequest.ProtoReflect.Descriptor instead.
func (*CreateChannelHandleRequest) GetChannelType ¶
func (x *CreateChannelHandleRequest) GetChannelType() data.ChannelHandle_ChannelType
func (*CreateChannelHandleRequest) ProtoMessage ¶
func (*CreateChannelHandleRequest) ProtoMessage()
func (*CreateChannelHandleRequest) ProtoReflect ¶
func (x *CreateChannelHandleRequest) ProtoReflect() protoreflect.Message
func (*CreateChannelHandleRequest) Reset ¶
func (x *CreateChannelHandleRequest) Reset()
func (*CreateChannelHandleRequest) String ¶
func (x *CreateChannelHandleRequest) String() string
type CreateChannelHandleResponse ¶
type CreateChannelHandleResponse struct { Channel *data.ChannelHandle `protobuf:"bytes,1,opt,name=channel,proto3" json:"channel,omitempty"` // contains filtered or unexported fields }
func (*CreateChannelHandleResponse) Descriptor
deprecated
func (*CreateChannelHandleResponse) Descriptor() ([]byte, []int)
Deprecated: Use CreateChannelHandleResponse.ProtoReflect.Descriptor instead.
func (*CreateChannelHandleResponse) GetChannel ¶
func (x *CreateChannelHandleResponse) GetChannel() *data.ChannelHandle
func (*CreateChannelHandleResponse) ProtoMessage ¶
func (*CreateChannelHandleResponse) ProtoMessage()
func (*CreateChannelHandleResponse) ProtoReflect ¶
func (x *CreateChannelHandleResponse) ProtoReflect() protoreflect.Message
func (*CreateChannelHandleResponse) Reset ¶
func (x *CreateChannelHandleResponse) Reset()
func (*CreateChannelHandleResponse) String ¶
func (x *CreateChannelHandleResponse) String() string
type DebugOptions ¶
type DebugOptions struct { // Show addresses of HLO ops in graph dump. XlaHloGraphAddresses bool `` /* 126-byte string literal not displayed */ // Instrument the computation to collect per-HLO cycle counts. XlaHloProfile bool `protobuf:"varint,9,opt,name=xla_hlo_profile,json=xlaHloProfile,proto3" json:"xla_hlo_profile,omitempty"` // List of HLO passes to disable/enable. These names must exactly match the // pass names as specified by the HloPassInterface::name() method. // // At least one of xla_disable_hlo_passes and xla_enable_hlo_passes_only must // be empty. XlaDisableHloPasses []string `protobuf:"bytes,30,rep,name=xla_disable_hlo_passes,json=xlaDisableHloPasses,proto3" json:"xla_disable_hlo_passes,omitempty"` XlaEnableHloPassesOnly []string `` /* 135-byte string literal not displayed */ // Disables all HLO passes. Notes that some passes are necessary for // correctness and the invariants that must be satisfied by "fully optimized" // HLO are different for different devices and may change over time. The only // "guarantee", such as it is, is that if you compile XLA and dump the // optimized HLO for some graph, you should be able to run it again on the // same device with the same build of XLA. XlaDisableAllHloPasses bool `` /* 136-byte string literal not displayed */ // Numerical optimization level for the XLA compiler backend; the specific // interpretation of this value is left to the backends. XlaBackendOptimizationLevel int32 `` /* 148-byte string literal not displayed */ // Embed the compiler IR as a string in the executable. XlaEmbedIrInExecutable bool `` /* 135-byte string literal not displayed */ // Eliminate implicit broadcasts when lowering user computations to HLO // instructions; use explicit broadcast instead. XlaEliminateHloImplicitBroadcast bool `` /* 165-byte string literal not displayed */ // When generating calls to Eigen in the CPU backend, use multi-threaded Eigen // mode. XlaCpuMultiThreadEigen bool `` /* 135-byte string literal not displayed */ // Path to directory with cuda/ptx tools and libraries. XlaGpuCudaDataDir string `protobuf:"bytes,61,opt,name=xla_gpu_cuda_data_dir,json=xlaGpuCudaDataDir,proto3" json:"xla_gpu_cuda_data_dir,omitempty"` // Enable flush-to-zero semantics in the GPU backend. XlaGpuFtz bool `protobuf:"varint,62,opt,name=xla_gpu_ftz,json=xlaGpuFtz,proto3" json:"xla_gpu_ftz,omitempty"` // Disable multi-streaming in the GPU backend. XlaGpuDisableMultiStreaming bool `` /* 150-byte string literal not displayed */ // Debugging feature: if enabled, the GPU backend will assign HLO operators to // randomly chosen streams. This is intended to trigger concurrency bugs. XlaGpuUseRandomStreams bool `` /* 136-byte string literal not displayed */ // If true, in LLVM-based backends, emit !alias.scope metadata in // generated IR. XlaLlvmEnableAliasScopeMetadata bool `` /* 164-byte string literal not displayed */ // If true, in LLVM-based backends, emit !noalias metadata in the // generated IR. XlaLlvmEnableNoaliasMetadata bool `` /* 153-byte string literal not displayed */ // If true, in LLVM-based backends, emit !invariant.load metadata in // the generated IR. XlaLlvmEnableInvariantLoadMetadata bool `` /* 173-byte string literal not displayed */ // If true, a set of expensive LLVM optimization passes will not be run. XlaLlvmDisableExpensivePasses bool `` /* 156-byte string literal not displayed */ // This is used by ClientLibraryTestBase::ComputeAndCompare*. If true, the // computation will run n! times with all permunations of layouts for the // output shape in rank n. For example, with a 3D shape, all permutations of // the set {0, 1, 2} are tried. XlaTestAllOutputLayouts bool `` /* 138-byte string literal not displayed */ // This is used by ClientLibraryTestBase::ComputeAndCompare*. If true, the // computation will run for all permunations of layouts of all input // arguments. For example, with 2 input arguments in 2D and 4D shapes, the // computation will run 2! * 4! times. XlaTestAllInputLayouts bool `` /* 135-byte string literal not displayed */ // Assign colors based on sharding information when generating the Graphviz // HLO graph. XlaHloGraphShardingColor bool `` /* 141-byte string literal not displayed */ // Generate calls to MKL-DNN in the CPU backend. XlaCpuUseMklDnn bool `protobuf:"varint,97,opt,name=xla_cpu_use_mkl_dnn,json=xlaCpuUseMklDnn,proto3" json:"xla_cpu_use_mkl_dnn,omitempty"` // Enable JitRt in the CPU backend. XlaCpuUseJitrt bool `protobuf:"varint,177,opt,name=xla_cpu_use_jitrt,json=xlaCpuUseJitrt,proto3" json:"xla_cpu_use_jitrt,omitempty"` // Maximum kernel unroll factor for the GPU backend. XlaGpuMaxKernelUnrollFactor int32 `` /* 152-byte string literal not displayed */ // When true, "unsafe" mathematical optimizations are enabled. These // transformations include but are not limited to: // // - Reducing the precision of operations (e.g. using an approximate sin // function, or transforming x/y into x * (1/y)). // - Assuming that operations never produce or consume NaN or +/- Inf (this // behavior can be adjusted using xla_cpu_fast_math_allow_{nans|infs}). // - Assuming that +0 and -0 are indistinguishable. XlaCpuEnableFastMath bool `` /* 129-byte string literal not displayed */ // When xla_cpu_enable_fast_math is true then this controls whether we allow // operations to produce NaNs. Ignored when xla_cpu_enable_fast_math is // false. XlaCpuFastMathHonorNans bool `` /* 141-byte string literal not displayed */ // When xla_cpu_enable_fast_math is true then this controls whether we allow // operations to produce infinites. Ignored when xla_cpu_enable_fast_math is // false. XlaCpuFastMathHonorInfs bool `` /* 141-byte string literal not displayed */ // When xla_cpu_enable_fast_math is true then this controls whether we forbid // to use the reciprocal of an argument instead of division. Ignored when // xla_cpu_enable_fast_math is false. XlaCpuFastMathHonorDivision bool `` /* 153-byte string literal not displayed */ // When xla_cpu_enable_fast_math is true then this controls whether we forbid // to approximate calculations for functions. Ignored when // xla_cpu_enable_fast_math is false. XlaCpuFastMathHonorFunctions bool `` /* 156-byte string literal not displayed */ // When false we lower the Minimum and Maximum hlos in the CPU backend such // that Min(NotNaN, NaN) = Min(NaN, NotNaN) = NaN. In other words, if flag // this is false we always propagate NaNs through Min and Max. // // Note, this does not correspond to the exact same behavior as the gpu flag // below! XlaCpuEnableFastMinMax bool `` /* 138-byte string literal not displayed */ // When true we lower the Minimum and Maximum hlos in the GPU backend such // that Min(NotNaN, NaN) = Min(NaN, NotNaN) = NotNaN. In other words, if flag // this is true we don't propagate NaNs through Min and Max. // // Note, this does not correspond to the exact same behavior as the cpu flag // above! XlaGpuEnableFastMinMax bool `` /* 138-byte string literal not displayed */ // Allows xla to increase the output precision of floating point operations. XlaAllowExcessPrecision bool `` /* 137-byte string literal not displayed */ // Crashes the program when any kind of verification fails, instead of just // logging the failures. One example is cross checking of convolution results // among different algorithms. XlaGpuCrashOnVerificationFailures bool `` /* 171-byte string literal not displayed */ // 0: Disable gemm and convolution autotuning. // 1: Enable autotuning, but disable correctness checking. // 2: Also set output buffers to random numbers during autotuning. // 3: Also reset output buffers to random numbers after autotuning each // // algorithm. // // 4+: Also check for correct outputs and for out-of-bounds reads/writes. // // Default: 4. XlaGpuAutotuneLevel int32 `protobuf:"varint,123,opt,name=xla_gpu_autotune_level,json=xlaGpuAutotuneLevel,proto3" json:"xla_gpu_autotune_level,omitempty"` // Force the host platform to pretend that there are these many host // "devices". All these devices are backed by the same threadpool. Defaults // to 1. // // Setting this to anything other than 1 can increase overhead from context // switching but we let the user override this behavior to help run tests on // the host that run models in parallel across multiple devices. XlaForceHostPlatformDeviceCount int32 `` /* 165-byte string literal not displayed */ // If set to true XLA:GPU invokes `ptxas` with -O0 (default is -O3). XlaGpuDisableGpuasmOptimizations bool `` /* 166-byte string literal not displayed */ XlaGpuShapeChecks DebugOptions_ShapeChecks `` /* 153-byte string literal not displayed */ // Enable MLIR-based lowering in XLA:CPU instead of LLVM emitters. XlaCpuEnableMlirLowering bool `` /* 142-byte string literal not displayed */ // If true, use MLIR instead of IR emitter to generate device code for // supported lmhlo.fusion ops. See xla::gpu::RewriteFusionOps() for details. XlaGpuEnableMlirLowering bool `` /* 142-byte string literal not displayed */ // Enable fast math with eigen in the HLO evaluator. XlaHloEvaluatorUseFastPath bool `` /* 150-byte string literal not displayed */ // Temporary option to allow support for both the R1 and the scalar index // versions of DynamicSlice and DynamicUpdateSlice. Only used for testing. XlaAllowScalarIndexDynamicOps bool `` /* 159-byte string literal not displayed */ // Option to emit a target-specific marker to indicate the start of a training // step. The location of the marker (if any) is determined by the option // value. XlaStepMarkerLocation DebugOptions_StepMarkerLocation `` /* 172-byte string literal not displayed */ // Directory to dump into. XlaDumpTo string `protobuf:"bytes,109,opt,name=xla_dump_to,json=xlaDumpTo,proto3" json:"xla_dump_to,omitempty"` // If specified, will only dump modules which match this regexp. XlaDumpHloModuleRe string `protobuf:"bytes,110,opt,name=xla_dump_hlo_module_re,json=xlaDumpHloModuleRe,proto3" json:"xla_dump_hlo_module_re,omitempty"` // If this flag is specified, will also dump HLO before and after passes that // match this regular expression. Set to .* to dump before/after all passes. XlaDumpHloPassRe string `protobuf:"bytes,111,opt,name=xla_dump_hlo_pass_re,json=xlaDumpHloPassRe,proto3" json:"xla_dump_hlo_pass_re,omitempty"` // Specifies the format that HLO is dumped in. Multiple of these may be // specified. XlaDumpHloAsText bool `protobuf:"varint,112,opt,name=xla_dump_hlo_as_text,json=xlaDumpHloAsText,proto3" json:"xla_dump_hlo_as_text,omitempty"` XlaDumpHloAsProto bool `protobuf:"varint,113,opt,name=xla_dump_hlo_as_proto,json=xlaDumpHloAsProto,proto3" json:"xla_dump_hlo_as_proto,omitempty"` XlaDumpHloAsDot bool `protobuf:"varint,114,opt,name=xla_dump_hlo_as_dot,json=xlaDumpHloAsDot,proto3" json:"xla_dump_hlo_as_dot,omitempty"` XlaDumpHloAsUrl bool `protobuf:"varint,115,opt,name=xla_dump_hlo_as_url,json=xlaDumpHloAsUrl,proto3" json:"xla_dump_hlo_as_url,omitempty"` // Dump HLO graphs as an HTML (DOT -> SVG inlined in HTML) XlaDumpHloAsHtml bool `protobuf:"varint,116,opt,name=xla_dump_hlo_as_html,json=xlaDumpHloAsHtml,proto3" json:"xla_dump_hlo_as_html,omitempty"` // Dump the visualization of the fusion progress. XlaDumpFusionVisualization bool `` /* 146-byte string literal not displayed */ // If true, every time an HLO module is run, we will dump an HloSnapshot // (essentially, a serialized module plus its inputs) to the --xla_dump_to // directory. XlaDumpHloSnapshots bool `protobuf:"varint,118,opt,name=xla_dump_hlo_snapshots,json=xlaDumpHloSnapshots,proto3" json:"xla_dump_hlo_snapshots,omitempty"` // Include a timestamp in the dumped filenames. XlaDumpIncludeTimestamp bool `` /* 137-byte string literal not displayed */ // Max number of hlo module dumps in a directory. Set to < 0 for unbounded. XlaDumpMaxHloModules int32 `` /* 130-byte string literal not displayed */ // Dump HloModuleMetadata as a text proto for each HLO module. XlaDumpModuleMetadata bool `` /* 131-byte string literal not displayed */ // GZip-compress protos dumped via --xla_dump_hlo_as_proto. XlaDumpCompressProtos bool `` /* 131-byte string literal not displayed */ // Dump HLO in long text format. Ignored unless xla_dump_hlo_as_text is true. XlaDumpHloAsLongText bool `` /* 132-byte string literal not displayed */ // Overrides for XLA GPU's convolution layout heuristic. XlaGpuForceConvNchw bool `` /* 127-byte string literal not displayed */ XlaGpuForceConvNhwc bool `` /* 127-byte string literal not displayed */ // Paths to files with ptx code. XlaGpuPtxFile []string `protobuf:"bytes,127,rep,name=xla_gpu_ptx_file,json=xlaGpuPtxFile,proto3" json:"xla_gpu_ptx_file,omitempty"` // Whether to dump llvm ir when compiling to ptx. XlaGpuDumpLlvmir bool `protobuf:"varint,155,opt,name=xla_gpu_dump_llvmir,json=xlaGpuDumpLlvmir,proto3" json:"xla_gpu_dump_llvmir,omitempty"` // Denylist for cuDNN convolutions. XlaGpuAlgorithmDenylistPath string `` /* 150-byte string literal not displayed */ // Debug options that trigger execution errors when NaN or Inf are detected. XlaTpuDetectNan bool `protobuf:"varint,135,opt,name=xla_tpu_detect_nan,json=xlaTpuDetectNan,proto3" json:"xla_tpu_detect_nan,omitempty"` XlaTpuDetectInf bool `protobuf:"varint,136,opt,name=xla_tpu_detect_inf,json=xlaTpuDetectInf,proto3" json:"xla_tpu_detect_inf,omitempty"` // True if TraceMe annotations are enabled for XLA:CPU. XlaCpuEnableXprofTraceme bool `` /* 142-byte string literal not displayed */ // It is usually preferable to not fallback to the driver; it can consume more // memory, or have bugs. XlaGpuUnsafeFallbackToDriverOnPtxasNotFound bool `` /* 209-byte string literal not displayed */ // Extra parameters to pass the GPU assembler. XlaGpuAsmExtraFlags string `` /* 126-byte string literal not displayed */ // Per-heap size constraint. New heaps will be created if per-heap max size is // reached. XlaMultiheapSizeConstraintPerHeap int32 `` /* 171-byte string literal not displayed */ // Enable detailed logging into vlog and xla dumping. If this is disabled, no // compilation summary will be printed in the end of computation and no hlo // modules will be dumped. XlaDetailedLoggingAndDumping bool `` /* 154-byte string literal not displayed */ // Overrides normal multi-threaded compilation settting to use this many // threads. Setting to 0 (the default value) means no enforcement. XlaGpuForceCompilationParallelism int32 `` /* 169-byte string literal not displayed */ // Guarantees run-to-run determinism. At present, the HLO ops Scatter and // SelectAndScatter do not have deterministic XLA:GPU implementations. // Compilation errors out if these ops are encountered. XlaGpuDeterministicOps bool `` /* 134-byte string literal not displayed */ // Simplify all Scatter ops to the form described in scatter_simplifier.h. XlaGpuSimplifyScatters bool `` /* 134-byte string literal not displayed */ // Simplify all Gather ops to the form described in gather_simplifier.h. XlaGpuSimplifyGathers bool `` /* 131-byte string literal not displayed */ // Paths to files with LLVM code. XlaGpuLlvmIrFile []string `protobuf:"bytes,150,rep,name=xla_gpu_llvm_ir_file,json=xlaGpuLlvmIrFile,proto3" json:"xla_gpu_llvm_ir_file,omitempty"` // Convert synchronous all-reduces ops into asynchronous. XlaGpuEnableAsyncAllReduce bool `` /* 150-byte string literal not displayed */ // Size threshold (in bytes) for the GPU all-reduce combiner. XlaGpuAllReduceCombineThresholdBytes int64 `` /* 182-byte string literal not displayed */ // Combine GPU all-reduces into a single operation over a contiguous buffer. XlaGpuAllReduceContiguous bool `` /* 145-byte string literal not displayed */ // Number of devices per host for first stage of BlueConnect decomposition // pass. The pass will attempt to decompose all-reduces ops into a // ReduceScatter-AllReduce-AllGather sequence, with the initial ReduceScatter // being performed over all of the devices in the same host. Set to < 1 to // disable all-reduce decomposition. XlaGpuAllReduceBlueconnectNumDevicesPerHost int32 `` /* 207-byte string literal not displayed */ // Whether to use the cuDNN frontend API for convolutions when possible. XlaGpuEnableCudnnFrontend bool `` /* 145-byte string literal not displayed */ // Disable dumping metadata in HLO dumps. XlaDumpDisableMetadata bool `` /* 134-byte string literal not displayed */ // If this flag is specified, will only dump HLO before and after passes in // the pass pipeline that matches this regular expression. Default empty value // enables dumping in all pipelines. XlaDumpHloPipelineRe string `` /* 129-byte string literal not displayed */ // If true, abort immediately when conv algorithm picker fails, rather than // logging a warning and proceeding with fallback. XlaGpuStrictConvAlgorithmPicker bool `` /* 165-byte string literal not displayed */ // If true, enable XLIR to compile gpu programs to TFRT BEF. XlaGpuBefExecutable bool `protobuf:"varint,161,opt,name=xla_gpu_bef_executable,json=xlaGpuBefExecutable,proto3" json:"xla_gpu_bef_executable,omitempty"` // If true, enable XLIR to compile thunks to TFRT BEF. // This flag has no effect when xla_gpu_bef_executable is true. XlaGpuBefThunk bool `protobuf:"varint,162,opt,name=xla_gpu_bef_thunk,json=xlaGpuBefThunk,proto3" json:"xla_gpu_bef_thunk,omitempty"` // If true, enable XLIR to compile gpu programs to JitRt. XlaGpuJitrtExecutable bool `` /* 131-byte string literal not displayed */ // Timeout in seconds before terminating jobs that are stuck in a NCCL // Rendezvous. Negative value disables the timeout and will not terminate. XlaGpuNcclTerminationTimeoutSeconds int64 `` /* 177-byte string literal not displayed */ // shared among multiple GPU executables. XlaGpuEnableSharedConstants bool `` /* 151-byte string literal not displayed */ // Whether to use cuBLASLt for GEMMs on GPUs. XlaGpuEnableCublaslt bool `` /* 128-byte string literal not displayed */ // Size threshold (in megabytes) for the GPU redzone scratch allocator. XlaGpuRedzoneScratchMaxMegabytes int64 `` /* 168-byte string literal not displayed */ // Allows all floating-point conversions to be simplified, including those // that affect the numerics. The `BFloat16Normalization` pass inserts many // `f32 -> bf16 -> f32` conversion pairs. These are not removed by the // `AlgebraicSimplifier`, as that will only simplify conversions that are // no-ops, e.g. `bf16 -> f32 -> bf16`. Removing these improves accuracy. XlaGpuSimplifyAllFpConversions bool `` /* 162-byte string literal not displayed */ // An experimental option to force all layouts present in the // after-optimizations HLO to be descending, e.g. // ShapeUtil::MakeShapeWithDescendingLayout is an identity on all // instructions. XlaGpuNormalizeLayouts bool `` /* 134-byte string literal not displayed */ // Generate calls to Arm Compute Library in the CPU backend. XlaCpuUseAcl bool `protobuf:"varint,174,opt,name=xla_cpu_use_acl,json=xlaCpuUseAcl,proto3" json:"xla_cpu_use_acl,omitempty"` // By default, XLA:CPU will run fp16 dot/conv as fp32, as this is generally // (much) faster on our hardware. Set this flag to disable this behavior. XlaCpuStrictDotConvMath bool `` /* 141-byte string literal not displayed */ // Extra options to pass to the compilation backend (e.g. LLVM); specific // interpretation of these values is left to the backend. XlaBackendExtraOptions map[string]string `` /* 221-byte string literal not displayed */ // contains filtered or unexported fields }
Debugging options for XLA. These options may change at any time - there are no guarantees about backward or forward compatibility for these fields.
func (*DebugOptions) Descriptor
deprecated
func (*DebugOptions) Descriptor() ([]byte, []int)
Deprecated: Use DebugOptions.ProtoReflect.Descriptor instead.
func (*DebugOptions) GetXlaAllowExcessPrecision ¶
func (x *DebugOptions) GetXlaAllowExcessPrecision() bool
func (*DebugOptions) GetXlaAllowScalarIndexDynamicOps ¶
func (x *DebugOptions) GetXlaAllowScalarIndexDynamicOps() bool
func (*DebugOptions) GetXlaBackendExtraOptions ¶
func (x *DebugOptions) GetXlaBackendExtraOptions() map[string]string
func (*DebugOptions) GetXlaBackendOptimizationLevel ¶
func (x *DebugOptions) GetXlaBackendOptimizationLevel() int32
func (*DebugOptions) GetXlaCpuEnableFastMath ¶
func (x *DebugOptions) GetXlaCpuEnableFastMath() bool
func (*DebugOptions) GetXlaCpuEnableFastMinMax ¶
func (x *DebugOptions) GetXlaCpuEnableFastMinMax() bool
func (*DebugOptions) GetXlaCpuEnableMlirLowering ¶
func (x *DebugOptions) GetXlaCpuEnableMlirLowering() bool
func (*DebugOptions) GetXlaCpuEnableXprofTraceme ¶
func (x *DebugOptions) GetXlaCpuEnableXprofTraceme() bool
func (*DebugOptions) GetXlaCpuFastMathHonorDivision ¶
func (x *DebugOptions) GetXlaCpuFastMathHonorDivision() bool
func (*DebugOptions) GetXlaCpuFastMathHonorFunctions ¶
func (x *DebugOptions) GetXlaCpuFastMathHonorFunctions() bool
func (*DebugOptions) GetXlaCpuFastMathHonorInfs ¶
func (x *DebugOptions) GetXlaCpuFastMathHonorInfs() bool
func (*DebugOptions) GetXlaCpuFastMathHonorNans ¶
func (x *DebugOptions) GetXlaCpuFastMathHonorNans() bool
func (*DebugOptions) GetXlaCpuMultiThreadEigen ¶
func (x *DebugOptions) GetXlaCpuMultiThreadEigen() bool
func (*DebugOptions) GetXlaCpuStrictDotConvMath ¶
func (x *DebugOptions) GetXlaCpuStrictDotConvMath() bool
func (*DebugOptions) GetXlaCpuUseAcl ¶
func (x *DebugOptions) GetXlaCpuUseAcl() bool
func (*DebugOptions) GetXlaCpuUseJitrt ¶
func (x *DebugOptions) GetXlaCpuUseJitrt() bool
func (*DebugOptions) GetXlaCpuUseMklDnn ¶
func (x *DebugOptions) GetXlaCpuUseMklDnn() bool
func (*DebugOptions) GetXlaDetailedLoggingAndDumping ¶
func (x *DebugOptions) GetXlaDetailedLoggingAndDumping() bool
func (*DebugOptions) GetXlaDisableAllHloPasses ¶
func (x *DebugOptions) GetXlaDisableAllHloPasses() bool
func (*DebugOptions) GetXlaDisableHloPasses ¶
func (x *DebugOptions) GetXlaDisableHloPasses() []string
func (*DebugOptions) GetXlaDumpCompressProtos ¶
func (x *DebugOptions) GetXlaDumpCompressProtos() bool
func (*DebugOptions) GetXlaDumpDisableMetadata ¶
func (x *DebugOptions) GetXlaDumpDisableMetadata() bool
func (*DebugOptions) GetXlaDumpFusionVisualization ¶
func (x *DebugOptions) GetXlaDumpFusionVisualization() bool
func (*DebugOptions) GetXlaDumpHloAsDot ¶
func (x *DebugOptions) GetXlaDumpHloAsDot() bool
func (*DebugOptions) GetXlaDumpHloAsHtml ¶
func (x *DebugOptions) GetXlaDumpHloAsHtml() bool
func (*DebugOptions) GetXlaDumpHloAsLongText ¶
func (x *DebugOptions) GetXlaDumpHloAsLongText() bool
func (*DebugOptions) GetXlaDumpHloAsProto ¶
func (x *DebugOptions) GetXlaDumpHloAsProto() bool
func (*DebugOptions) GetXlaDumpHloAsText ¶
func (x *DebugOptions) GetXlaDumpHloAsText() bool
func (*DebugOptions) GetXlaDumpHloAsUrl ¶
func (x *DebugOptions) GetXlaDumpHloAsUrl() bool
func (*DebugOptions) GetXlaDumpHloModuleRe ¶
func (x *DebugOptions) GetXlaDumpHloModuleRe() string
func (*DebugOptions) GetXlaDumpHloPassRe ¶
func (x *DebugOptions) GetXlaDumpHloPassRe() string
func (*DebugOptions) GetXlaDumpHloPipelineRe ¶
func (x *DebugOptions) GetXlaDumpHloPipelineRe() string
func (*DebugOptions) GetXlaDumpHloSnapshots ¶
func (x *DebugOptions) GetXlaDumpHloSnapshots() bool
func (*DebugOptions) GetXlaDumpIncludeTimestamp ¶
func (x *DebugOptions) GetXlaDumpIncludeTimestamp() bool
func (*DebugOptions) GetXlaDumpMaxHloModules ¶
func (x *DebugOptions) GetXlaDumpMaxHloModules() int32
func (*DebugOptions) GetXlaDumpModuleMetadata ¶
func (x *DebugOptions) GetXlaDumpModuleMetadata() bool
func (*DebugOptions) GetXlaDumpTo ¶
func (x *DebugOptions) GetXlaDumpTo() string
func (*DebugOptions) GetXlaEliminateHloImplicitBroadcast ¶
func (x *DebugOptions) GetXlaEliminateHloImplicitBroadcast() bool
func (*DebugOptions) GetXlaEmbedIrInExecutable ¶
func (x *DebugOptions) GetXlaEmbedIrInExecutable() bool
func (*DebugOptions) GetXlaEnableHloPassesOnly ¶
func (x *DebugOptions) GetXlaEnableHloPassesOnly() []string
func (*DebugOptions) GetXlaForceHostPlatformDeviceCount ¶
func (x *DebugOptions) GetXlaForceHostPlatformDeviceCount() int32
func (*DebugOptions) GetXlaGpuAlgorithmDenylistPath ¶
func (x *DebugOptions) GetXlaGpuAlgorithmDenylistPath() string
func (*DebugOptions) GetXlaGpuAllReduceBlueconnectNumDevicesPerHost ¶
func (x *DebugOptions) GetXlaGpuAllReduceBlueconnectNumDevicesPerHost() int32
func (*DebugOptions) GetXlaGpuAllReduceCombineThresholdBytes ¶
func (x *DebugOptions) GetXlaGpuAllReduceCombineThresholdBytes() int64
func (*DebugOptions) GetXlaGpuAllReduceContiguous ¶
func (x *DebugOptions) GetXlaGpuAllReduceContiguous() bool
func (*DebugOptions) GetXlaGpuAsmExtraFlags ¶
func (x *DebugOptions) GetXlaGpuAsmExtraFlags() string
func (*DebugOptions) GetXlaGpuAutotuneLevel ¶
func (x *DebugOptions) GetXlaGpuAutotuneLevel() int32
func (*DebugOptions) GetXlaGpuBefExecutable ¶
func (x *DebugOptions) GetXlaGpuBefExecutable() bool
func (*DebugOptions) GetXlaGpuBefThunk ¶
func (x *DebugOptions) GetXlaGpuBefThunk() bool
func (*DebugOptions) GetXlaGpuCrashOnVerificationFailures ¶
func (x *DebugOptions) GetXlaGpuCrashOnVerificationFailures() bool
func (*DebugOptions) GetXlaGpuCudaDataDir ¶
func (x *DebugOptions) GetXlaGpuCudaDataDir() string
func (*DebugOptions) GetXlaGpuDeterministicOps ¶
func (x *DebugOptions) GetXlaGpuDeterministicOps() bool
func (*DebugOptions) GetXlaGpuDisableGpuasmOptimizations ¶
func (x *DebugOptions) GetXlaGpuDisableGpuasmOptimizations() bool
func (*DebugOptions) GetXlaGpuDisableMultiStreaming ¶
func (x *DebugOptions) GetXlaGpuDisableMultiStreaming() bool
func (*DebugOptions) GetXlaGpuDumpLlvmir ¶
func (x *DebugOptions) GetXlaGpuDumpLlvmir() bool
func (*DebugOptions) GetXlaGpuEnableAsyncAllReduce ¶
func (x *DebugOptions) GetXlaGpuEnableAsyncAllReduce() bool
func (*DebugOptions) GetXlaGpuEnableCublaslt ¶
func (x *DebugOptions) GetXlaGpuEnableCublaslt() bool
func (*DebugOptions) GetXlaGpuEnableCudnnFrontend ¶
func (x *DebugOptions) GetXlaGpuEnableCudnnFrontend() bool
func (*DebugOptions) GetXlaGpuEnableFastMinMax ¶
func (x *DebugOptions) GetXlaGpuEnableFastMinMax() bool
func (*DebugOptions) GetXlaGpuEnableMlirLowering ¶
func (x *DebugOptions) GetXlaGpuEnableMlirLowering() bool
func (*DebugOptions) GetXlaGpuEnableSharedConstants ¶
func (x *DebugOptions) GetXlaGpuEnableSharedConstants() bool
func (*DebugOptions) GetXlaGpuForceCompilationParallelism ¶
func (x *DebugOptions) GetXlaGpuForceCompilationParallelism() int32
func (*DebugOptions) GetXlaGpuForceConvNchw ¶
func (x *DebugOptions) GetXlaGpuForceConvNchw() bool
func (*DebugOptions) GetXlaGpuForceConvNhwc ¶
func (x *DebugOptions) GetXlaGpuForceConvNhwc() bool
func (*DebugOptions) GetXlaGpuFtz ¶
func (x *DebugOptions) GetXlaGpuFtz() bool
func (*DebugOptions) GetXlaGpuJitrtExecutable ¶
func (x *DebugOptions) GetXlaGpuJitrtExecutable() bool
func (*DebugOptions) GetXlaGpuLlvmIrFile ¶
func (x *DebugOptions) GetXlaGpuLlvmIrFile() []string
func (*DebugOptions) GetXlaGpuMaxKernelUnrollFactor ¶
func (x *DebugOptions) GetXlaGpuMaxKernelUnrollFactor() int32
func (*DebugOptions) GetXlaGpuNcclTerminationTimeoutSeconds ¶
func (x *DebugOptions) GetXlaGpuNcclTerminationTimeoutSeconds() int64
func (*DebugOptions) GetXlaGpuNormalizeLayouts ¶
func (x *DebugOptions) GetXlaGpuNormalizeLayouts() bool
func (*DebugOptions) GetXlaGpuPtxFile ¶
func (x *DebugOptions) GetXlaGpuPtxFile() []string
func (*DebugOptions) GetXlaGpuRedzoneScratchMaxMegabytes ¶
func (x *DebugOptions) GetXlaGpuRedzoneScratchMaxMegabytes() int64
func (*DebugOptions) GetXlaGpuShapeChecks ¶
func (x *DebugOptions) GetXlaGpuShapeChecks() DebugOptions_ShapeChecks
func (*DebugOptions) GetXlaGpuSimplifyAllFpConversions ¶
func (x *DebugOptions) GetXlaGpuSimplifyAllFpConversions() bool
func (*DebugOptions) GetXlaGpuSimplifyGathers ¶
func (x *DebugOptions) GetXlaGpuSimplifyGathers() bool
func (*DebugOptions) GetXlaGpuSimplifyScatters ¶
func (x *DebugOptions) GetXlaGpuSimplifyScatters() bool
func (*DebugOptions) GetXlaGpuStrictConvAlgorithmPicker ¶
func (x *DebugOptions) GetXlaGpuStrictConvAlgorithmPicker() bool
func (*DebugOptions) GetXlaGpuUnsafeFallbackToDriverOnPtxasNotFound ¶
func (x *DebugOptions) GetXlaGpuUnsafeFallbackToDriverOnPtxasNotFound() bool
func (*DebugOptions) GetXlaGpuUseRandomStreams ¶
func (x *DebugOptions) GetXlaGpuUseRandomStreams() bool
func (*DebugOptions) GetXlaHloEvaluatorUseFastPath ¶
func (x *DebugOptions) GetXlaHloEvaluatorUseFastPath() bool
func (*DebugOptions) GetXlaHloGraphAddresses ¶
func (x *DebugOptions) GetXlaHloGraphAddresses() bool
func (*DebugOptions) GetXlaHloGraphShardingColor ¶
func (x *DebugOptions) GetXlaHloGraphShardingColor() bool
func (*DebugOptions) GetXlaHloProfile ¶
func (x *DebugOptions) GetXlaHloProfile() bool
func (*DebugOptions) GetXlaLlvmDisableExpensivePasses ¶
func (x *DebugOptions) GetXlaLlvmDisableExpensivePasses() bool
func (*DebugOptions) GetXlaLlvmEnableAliasScopeMetadata ¶
func (x *DebugOptions) GetXlaLlvmEnableAliasScopeMetadata() bool
func (*DebugOptions) GetXlaLlvmEnableInvariantLoadMetadata ¶
func (x *DebugOptions) GetXlaLlvmEnableInvariantLoadMetadata() bool
func (*DebugOptions) GetXlaLlvmEnableNoaliasMetadata ¶
func (x *DebugOptions) GetXlaLlvmEnableNoaliasMetadata() bool
func (*DebugOptions) GetXlaMultiheapSizeConstraintPerHeap ¶
func (x *DebugOptions) GetXlaMultiheapSizeConstraintPerHeap() int32
func (*DebugOptions) GetXlaStepMarkerLocation ¶
func (x *DebugOptions) GetXlaStepMarkerLocation() DebugOptions_StepMarkerLocation
func (*DebugOptions) GetXlaTestAllInputLayouts ¶
func (x *DebugOptions) GetXlaTestAllInputLayouts() bool
func (*DebugOptions) GetXlaTestAllOutputLayouts ¶
func (x *DebugOptions) GetXlaTestAllOutputLayouts() bool
func (*DebugOptions) GetXlaTpuDetectInf ¶
func (x *DebugOptions) GetXlaTpuDetectInf() bool
func (*DebugOptions) GetXlaTpuDetectNan ¶
func (x *DebugOptions) GetXlaTpuDetectNan() bool
func (*DebugOptions) ProtoMessage ¶
func (*DebugOptions) ProtoMessage()
func (*DebugOptions) ProtoReflect ¶
func (x *DebugOptions) ProtoReflect() protoreflect.Message
func (*DebugOptions) Reset ¶
func (x *DebugOptions) Reset()
func (*DebugOptions) String ¶
func (x *DebugOptions) String() string
type DebugOptions_ShapeChecks ¶
type DebugOptions_ShapeChecks int32
const ( // Do not insert any shape checks for dynamically shaped operations; output // buffers might contain garbage data if shapes don't match. DebugOptions_IGNORE DebugOptions_ShapeChecks = 0 // Check shapes at runtime, will insert an extra synchronization if shapes // cannot be proven correct at compile time. DebugOptions_RUNTIME DebugOptions_ShapeChecks = 1 // Will refuse to compile any program where shape correctness can not be // established at compile time. DebugOptions_COMPILE_TIME DebugOptions_ShapeChecks = 2 )
func (DebugOptions_ShapeChecks) Descriptor ¶
func (DebugOptions_ShapeChecks) Descriptor() protoreflect.EnumDescriptor
func (DebugOptions_ShapeChecks) Enum ¶
func (x DebugOptions_ShapeChecks) Enum() *DebugOptions_ShapeChecks
func (DebugOptions_ShapeChecks) EnumDescriptor
deprecated
func (DebugOptions_ShapeChecks) EnumDescriptor() ([]byte, []int)
Deprecated: Use DebugOptions_ShapeChecks.Descriptor instead.
func (DebugOptions_ShapeChecks) Number ¶
func (x DebugOptions_ShapeChecks) Number() protoreflect.EnumNumber
func (DebugOptions_ShapeChecks) String ¶
func (x DebugOptions_ShapeChecks) String() string
func (DebugOptions_ShapeChecks) Type ¶
func (DebugOptions_ShapeChecks) Type() protoreflect.EnumType
type DebugOptions_StepMarkerLocation ¶
type DebugOptions_StepMarkerLocation int32
const ( // Generate a step marker at the program entry. This handles the case where // each step is done by one or multiple program execution(s). Only the first // program will be tagged for generating a step marker at the program entry. // This is the default. DebugOptions_STEP_MARK_AT_ENTRY DebugOptions_StepMarkerLocation = 0 // Generate a step marker at each iteration of the top level while loop, // which is assumed to be a training loop. DebugOptions_STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP DebugOptions_StepMarkerLocation = 1 // Generate a step marker at each iteration of the second level while loops, // which is assumed to be a training or eval loop. DebugOptions_STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP DebugOptions_StepMarkerLocation = 3 // No step marker generated. DebugOptions_STEP_MARK_NONE DebugOptions_StepMarkerLocation = 2 )
func (DebugOptions_StepMarkerLocation) Descriptor ¶
func (DebugOptions_StepMarkerLocation) Descriptor() protoreflect.EnumDescriptor
func (DebugOptions_StepMarkerLocation) Enum ¶
func (x DebugOptions_StepMarkerLocation) Enum() *DebugOptions_StepMarkerLocation
func (DebugOptions_StepMarkerLocation) EnumDescriptor
deprecated
func (DebugOptions_StepMarkerLocation) EnumDescriptor() ([]byte, []int)
Deprecated: Use DebugOptions_StepMarkerLocation.Descriptor instead.
func (DebugOptions_StepMarkerLocation) Number ¶
func (x DebugOptions_StepMarkerLocation) Number() protoreflect.EnumNumber
func (DebugOptions_StepMarkerLocation) String ¶
func (x DebugOptions_StepMarkerLocation) String() string
func (DebugOptions_StepMarkerLocation) Type ¶
func (DebugOptions_StepMarkerLocation) Type() protoreflect.EnumType
type DeconstructTupleRequest ¶
type DeconstructTupleRequest struct { TupleHandle *data.GlobalDataHandle `protobuf:"bytes,2,opt,name=tuple_handle,json=tupleHandle,proto3" json:"tuple_handle,omitempty"` // contains filtered or unexported fields }
func (*DeconstructTupleRequest) Descriptor
deprecated
func (*DeconstructTupleRequest) Descriptor() ([]byte, []int)
Deprecated: Use DeconstructTupleRequest.ProtoReflect.Descriptor instead.
func (*DeconstructTupleRequest) GetTupleHandle ¶
func (x *DeconstructTupleRequest) GetTupleHandle() *data.GlobalDataHandle
func (*DeconstructTupleRequest) ProtoMessage ¶
func (*DeconstructTupleRequest) ProtoMessage()
func (*DeconstructTupleRequest) ProtoReflect ¶
func (x *DeconstructTupleRequest) ProtoReflect() protoreflect.Message
func (*DeconstructTupleRequest) Reset ¶
func (x *DeconstructTupleRequest) Reset()
func (*DeconstructTupleRequest) String ¶
func (x *DeconstructTupleRequest) String() string
type DeconstructTupleResponse ¶
type DeconstructTupleResponse struct { ElementHandles []*data.GlobalDataHandle `protobuf:"bytes,1,rep,name=element_handles,json=elementHandles,proto3" json:"element_handles,omitempty"` // contains filtered or unexported fields }
func (*DeconstructTupleResponse) Descriptor
deprecated
func (*DeconstructTupleResponse) Descriptor() ([]byte, []int)
Deprecated: Use DeconstructTupleResponse.ProtoReflect.Descriptor instead.
func (*DeconstructTupleResponse) GetElementHandles ¶
func (x *DeconstructTupleResponse) GetElementHandles() []*data.GlobalDataHandle
func (*DeconstructTupleResponse) ProtoMessage ¶
func (*DeconstructTupleResponse) ProtoMessage()
func (*DeconstructTupleResponse) ProtoReflect ¶
func (x *DeconstructTupleResponse) ProtoReflect() protoreflect.Message
func (*DeconstructTupleResponse) Reset ¶
func (x *DeconstructTupleResponse) Reset()
func (*DeconstructTupleResponse) String ¶
func (x *DeconstructTupleResponse) String() string
type ExecuteGraphParallelRequest ¶
type ExecuteGraphParallelRequest struct { Requests []*ExecuteGraphRequest `protobuf:"bytes,1,rep,name=requests,proto3" json:"requests,omitempty"` // contains filtered or unexported fields }
func (*ExecuteGraphParallelRequest) Descriptor
deprecated
func (*ExecuteGraphParallelRequest) Descriptor() ([]byte, []int)
Deprecated: Use ExecuteGraphParallelRequest.ProtoReflect.Descriptor instead.
func (*ExecuteGraphParallelRequest) GetRequests ¶
func (x *ExecuteGraphParallelRequest) GetRequests() []*ExecuteGraphRequest
func (*ExecuteGraphParallelRequest) ProtoMessage ¶
func (*ExecuteGraphParallelRequest) ProtoMessage()
func (*ExecuteGraphParallelRequest) ProtoReflect ¶
func (x *ExecuteGraphParallelRequest) ProtoReflect() protoreflect.Message
func (*ExecuteGraphParallelRequest) Reset ¶
func (x *ExecuteGraphParallelRequest) Reset()
func (*ExecuteGraphParallelRequest) String ¶
func (x *ExecuteGraphParallelRequest) String() string
type ExecuteGraphRequest ¶
type ExecuteGraphRequest struct { Computation *service.HloModuleProto `protobuf:"bytes,1,opt,name=computation,proto3" json:"computation,omitempty"` Arguments []*data.GlobalDataHandle `protobuf:"bytes,2,rep,name=arguments,proto3" json:"arguments,omitempty"` // Options that affect how XLA compiles and runs code to service this request. ExecutionOptions *ExecutionOptions `protobuf:"bytes,3,opt,name=execution_options,json=executionOptions,proto3" json:"execution_options,omitempty"` // contains filtered or unexported fields }
TODO(b/118493728): Remove this and ExecuteGraphParallelRequest and replace the uses with calls to Compile and Execute.
func (*ExecuteGraphRequest) Descriptor
deprecated
func (*ExecuteGraphRequest) Descriptor() ([]byte, []int)
Deprecated: Use ExecuteGraphRequest.ProtoReflect.Descriptor instead.
func (*ExecuteGraphRequest) GetArguments ¶
func (x *ExecuteGraphRequest) GetArguments() []*data.GlobalDataHandle
func (*ExecuteGraphRequest) GetComputation ¶
func (x *ExecuteGraphRequest) GetComputation() *service.HloModuleProto
func (*ExecuteGraphRequest) GetExecutionOptions ¶
func (x *ExecuteGraphRequest) GetExecutionOptions() *ExecutionOptions
func (*ExecuteGraphRequest) ProtoMessage ¶
func (*ExecuteGraphRequest) ProtoMessage()
func (*ExecuteGraphRequest) ProtoReflect ¶
func (x *ExecuteGraphRequest) ProtoReflect() protoreflect.Message
func (*ExecuteGraphRequest) Reset ¶
func (x *ExecuteGraphRequest) Reset()
func (*ExecuteGraphRequest) String ¶
func (x *ExecuteGraphRequest) String() string
type ExecuteParallelResponse ¶
type ExecuteParallelResponse struct { Responses []*ExecuteResponse `protobuf:"bytes,1,rep,name=responses,proto3" json:"responses,omitempty"` // contains filtered or unexported fields }
func (*ExecuteParallelResponse) Descriptor
deprecated
func (*ExecuteParallelResponse) Descriptor() ([]byte, []int)
Deprecated: Use ExecuteParallelResponse.ProtoReflect.Descriptor instead.
func (*ExecuteParallelResponse) GetResponses ¶
func (x *ExecuteParallelResponse) GetResponses() []*ExecuteResponse
func (*ExecuteParallelResponse) ProtoMessage ¶
func (*ExecuteParallelResponse) ProtoMessage()
func (*ExecuteParallelResponse) ProtoReflect ¶
func (x *ExecuteParallelResponse) ProtoReflect() protoreflect.Message
func (*ExecuteParallelResponse) Reset ¶
func (x *ExecuteParallelResponse) Reset()
func (*ExecuteParallelResponse) String ¶
func (x *ExecuteParallelResponse) String() string
type ExecuteRequest ¶
type ExecuteRequest struct { Handle *data.ExecutionHandle `protobuf:"bytes,1,opt,name=handle,proto3" json:"handle,omitempty"` // The shape and layout of the arguments must be the same as the those of the // executable's parameters. Arguments []*data.GlobalDataHandle `protobuf:"bytes,2,rep,name=arguments,proto3" json:"arguments,omitempty"` // contains filtered or unexported fields }
func (*ExecuteRequest) Descriptor
deprecated
func (*ExecuteRequest) Descriptor() ([]byte, []int)
Deprecated: Use ExecuteRequest.ProtoReflect.Descriptor instead.
func (*ExecuteRequest) GetArguments ¶
func (x *ExecuteRequest) GetArguments() []*data.GlobalDataHandle
func (*ExecuteRequest) GetHandle ¶
func (x *ExecuteRequest) GetHandle() *data.ExecutionHandle
func (*ExecuteRequest) ProtoMessage ¶
func (*ExecuteRequest) ProtoMessage()
func (*ExecuteRequest) ProtoReflect ¶
func (x *ExecuteRequest) ProtoReflect() protoreflect.Message
func (*ExecuteRequest) Reset ¶
func (x *ExecuteRequest) Reset()
func (*ExecuteRequest) String ¶
func (x *ExecuteRequest) String() string
type ExecuteResponse ¶
type ExecuteResponse struct { Output *data.GlobalDataHandle `protobuf:"bytes,1,opt,name=output,proto3" json:"output,omitempty"` Profile *data.ExecutionProfile `protobuf:"bytes,2,opt,name=profile,proto3" json:"profile,omitempty"` // contains filtered or unexported fields }
func (*ExecuteResponse) Descriptor
deprecated
func (*ExecuteResponse) Descriptor() ([]byte, []int)
Deprecated: Use ExecuteResponse.ProtoReflect.Descriptor instead.
func (*ExecuteResponse) GetOutput ¶
func (x *ExecuteResponse) GetOutput() *data.GlobalDataHandle
func (*ExecuteResponse) GetProfile ¶
func (x *ExecuteResponse) GetProfile() *data.ExecutionProfile
func (*ExecuteResponse) ProtoMessage ¶
func (*ExecuteResponse) ProtoMessage()
func (*ExecuteResponse) ProtoReflect ¶
func (x *ExecuteResponse) ProtoReflect() protoreflect.Message
func (*ExecuteResponse) Reset ¶
func (x *ExecuteResponse) Reset()
func (*ExecuteResponse) String ¶
func (x *ExecuteResponse) String() string
type ExecutionOptions ¶
type ExecutionOptions struct { // This optional field's layout is used as a hint when storing the output of // this computation. Subsequent transfers of this output array to the client // may be faster when using this layout. // // We use a Shape here to accommodate computations that return a tuple. ShapeWithOutputLayout *data.ShapeProto `` /* 128-byte string literal not displayed */ // Used to seed random-number generators used in this computation. If this is // 0, we generate a seed ourselves. // // TODO(b/32083678): Changing the seed unnecessarily forces a recompilation. Seed uint64 `protobuf:"varint,3,opt,name=seed,proto3" json:"seed,omitempty"` DebugOptions *DebugOptions `protobuf:"bytes,4,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"` // This optional field specifies a particular set of devices to run the // computation on. The computation will be partitioned across these devices. // If not provided, the default device will be chosen. DeviceHandles []*data.DeviceHandle `protobuf:"bytes,5,rep,name=device_handles,json=deviceHandles,proto3" json:"device_handles,omitempty"` // Number of replicas of the computation to run. If zero, uses the default // number of replicas for the XLA service. NumReplicas int32 `protobuf:"varint,6,opt,name=num_replicas,json=numReplicas,proto3" json:"num_replicas,omitempty"` // This optional field specifies the device assignment if known at compile // time. DeviceAssignment *data.DeviceAssignmentProto `protobuf:"bytes,7,opt,name=device_assignment,json=deviceAssignment,proto3" json:"device_assignment,omitempty"` // Alias input and output buffers for parameters that are passed-through XLA // modules without being changed. AliasPassthroughParams bool `` /* 130-byte string literal not displayed */ // Number of partitions of the computation to run (model parallelism). // If zero, uses the default number of partitions for the XLA service. NumPartitions int32 `protobuf:"varint,9,opt,name=num_partitions,json=numPartitions,proto3" json:"num_partitions,omitempty"` // Used to identify a set of programs that should be launch together. LaunchId int32 `protobuf:"varint,10,opt,name=launch_id,json=launchId,proto3" json:"launch_id,omitempty"` // Indicates whether to use SPMD (true) or MPMD (false) partitioning when // num_partitions > 1 and XLA is requested to partition the input program. UseSpmdPartitioning bool `protobuf:"varint,11,opt,name=use_spmd_partitioning,json=useSpmdPartitioning,proto3" json:"use_spmd_partitioning,omitempty"` // Whether to automatically generate XLA shardings for SPMD partitioner. UseAutoSpmdPartitioning bool `` /* 136-byte string literal not displayed */ // Device mesh shape used to create the sharding search space when // use_auto_spmd_partitioning=true. AutoSpmdPartitioningMeshShape []int64 `` /* 163-byte string literal not displayed */ // Device mesh ids compatible with the above mesh_shape used when // use_auto_spmd_partitioning=true. AutoSpmdPartitioningMeshIds []int64 `` /* 157-byte string literal not displayed */ // If set, deduplicate hlo into function calls to reduce binary size. Only // works on TPU. DeduplicateHlo bool `protobuf:"varint,12,opt,name=deduplicate_hlo,json=deduplicateHlo,proto3" json:"deduplicate_hlo,omitempty"` // Allows sharding propagation to propagate to the outputs. This changes the // output shape of the computation (which is undesirable), but it can be used // to allow to run partial compilation to determine what would be the output // sharding of a computation if XLA would be allowed to propagate the sharding // which can be used by higher level framework as a way to query intermediate // sharding of operations when multiple computation would be chained and // merged together. AllowSpmdShardingPropagationToOutput bool `` /* 179-byte string literal not displayed */ // contains filtered or unexported fields }
These settings control how XLA compiles and/or runs code. Not all settings will have an effect on every platform.
When adding new fields, keep in mind that boolean fields default to false.
func (*ExecutionOptions) Descriptor
deprecated
func (*ExecutionOptions) Descriptor() ([]byte, []int)
Deprecated: Use ExecutionOptions.ProtoReflect.Descriptor instead.
func (*ExecutionOptions) GetAliasPassthroughParams ¶
func (x *ExecutionOptions) GetAliasPassthroughParams() bool
func (*ExecutionOptions) GetAllowSpmdShardingPropagationToOutput ¶
func (x *ExecutionOptions) GetAllowSpmdShardingPropagationToOutput() bool
func (*ExecutionOptions) GetAutoSpmdPartitioningMeshIds ¶
func (x *ExecutionOptions) GetAutoSpmdPartitioningMeshIds() []int64
func (*ExecutionOptions) GetAutoSpmdPartitioningMeshShape ¶
func (x *ExecutionOptions) GetAutoSpmdPartitioningMeshShape() []int64
func (*ExecutionOptions) GetDebugOptions ¶
func (x *ExecutionOptions) GetDebugOptions() *DebugOptions
func (*ExecutionOptions) GetDeduplicateHlo ¶
func (x *ExecutionOptions) GetDeduplicateHlo() bool
func (*ExecutionOptions) GetDeviceAssignment ¶
func (x *ExecutionOptions) GetDeviceAssignment() *data.DeviceAssignmentProto
func (*ExecutionOptions) GetDeviceHandles ¶
func (x *ExecutionOptions) GetDeviceHandles() []*data.DeviceHandle
func (*ExecutionOptions) GetLaunchId ¶
func (x *ExecutionOptions) GetLaunchId() int32
func (*ExecutionOptions) GetNumPartitions ¶
func (x *ExecutionOptions) GetNumPartitions() int32
func (*ExecutionOptions) GetNumReplicas ¶
func (x *ExecutionOptions) GetNumReplicas() int32
func (*ExecutionOptions) GetSeed ¶
func (x *ExecutionOptions) GetSeed() uint64
func (*ExecutionOptions) GetShapeWithOutputLayout ¶
func (x *ExecutionOptions) GetShapeWithOutputLayout() *data.ShapeProto
func (*ExecutionOptions) GetUseAutoSpmdPartitioning ¶
func (x *ExecutionOptions) GetUseAutoSpmdPartitioning() bool
func (*ExecutionOptions) GetUseSpmdPartitioning ¶
func (x *ExecutionOptions) GetUseSpmdPartitioning() bool
func (*ExecutionOptions) ProtoMessage ¶
func (*ExecutionOptions) ProtoMessage()
func (*ExecutionOptions) ProtoReflect ¶
func (x *ExecutionOptions) ProtoReflect() protoreflect.Message
func (*ExecutionOptions) Reset ¶
func (x *ExecutionOptions) Reset()
func (*ExecutionOptions) String ¶
func (x *ExecutionOptions) String() string
type GetDeviceHandlesRequest ¶
type GetDeviceHandlesRequest struct { DeviceCount int64 `protobuf:"varint,1,opt,name=device_count,json=deviceCount,proto3" json:"device_count,omitempty"` // contains filtered or unexported fields }
func (*GetDeviceHandlesRequest) Descriptor
deprecated
func (*GetDeviceHandlesRequest) Descriptor() ([]byte, []int)
Deprecated: Use GetDeviceHandlesRequest.ProtoReflect.Descriptor instead.
func (*GetDeviceHandlesRequest) GetDeviceCount ¶
func (x *GetDeviceHandlesRequest) GetDeviceCount() int64
func (*GetDeviceHandlesRequest) ProtoMessage ¶
func (*GetDeviceHandlesRequest) ProtoMessage()
func (*GetDeviceHandlesRequest) ProtoReflect ¶
func (x *GetDeviceHandlesRequest) ProtoReflect() protoreflect.Message
func (*GetDeviceHandlesRequest) Reset ¶
func (x *GetDeviceHandlesRequest) Reset()
func (*GetDeviceHandlesRequest) String ¶
func (x *GetDeviceHandlesRequest) String() string
type GetDeviceHandlesResponse ¶
type GetDeviceHandlesResponse struct { DeviceHandles []*data.DeviceHandle `protobuf:"bytes,1,rep,name=device_handles,json=deviceHandles,proto3" json:"device_handles,omitempty"` // contains filtered or unexported fields }
func (*GetDeviceHandlesResponse) Descriptor
deprecated
func (*GetDeviceHandlesResponse) Descriptor() ([]byte, []int)
Deprecated: Use GetDeviceHandlesResponse.ProtoReflect.Descriptor instead.
func (*GetDeviceHandlesResponse) GetDeviceHandles ¶
func (x *GetDeviceHandlesResponse) GetDeviceHandles() []*data.DeviceHandle
func (*GetDeviceHandlesResponse) ProtoMessage ¶
func (*GetDeviceHandlesResponse) ProtoMessage()
func (*GetDeviceHandlesResponse) ProtoReflect ¶
func (x *GetDeviceHandlesResponse) ProtoReflect() protoreflect.Message
func (*GetDeviceHandlesResponse) Reset ¶
func (x *GetDeviceHandlesResponse) Reset()
func (*GetDeviceHandlesResponse) String ¶
func (x *GetDeviceHandlesResponse) String() string
type GetShapeRequest ¶
type GetShapeRequest struct { Data *data.GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"` // contains filtered or unexported fields }
func (*GetShapeRequest) Descriptor
deprecated
func (*GetShapeRequest) Descriptor() ([]byte, []int)
Deprecated: Use GetShapeRequest.ProtoReflect.Descriptor instead.
func (*GetShapeRequest) GetData ¶
func (x *GetShapeRequest) GetData() *data.GlobalDataHandle
func (*GetShapeRequest) ProtoMessage ¶
func (*GetShapeRequest) ProtoMessage()
func (*GetShapeRequest) ProtoReflect ¶
func (x *GetShapeRequest) ProtoReflect() protoreflect.Message
func (*GetShapeRequest) Reset ¶
func (x *GetShapeRequest) Reset()
func (*GetShapeRequest) String ¶
func (x *GetShapeRequest) String() string
type GetShapeResponse ¶
type GetShapeResponse struct { Shape *data.ShapeProto `protobuf:"bytes,1,opt,name=shape,proto3" json:"shape,omitempty"` // contains filtered or unexported fields }
func (*GetShapeResponse) Descriptor
deprecated
func (*GetShapeResponse) Descriptor() ([]byte, []int)
Deprecated: Use GetShapeResponse.ProtoReflect.Descriptor instead.
func (*GetShapeResponse) GetShape ¶
func (x *GetShapeResponse) GetShape() *data.ShapeProto
func (*GetShapeResponse) ProtoMessage ¶
func (*GetShapeResponse) ProtoMessage()
func (*GetShapeResponse) ProtoReflect ¶
func (x *GetShapeResponse) ProtoReflect() protoreflect.Message
func (*GetShapeResponse) Reset ¶
func (x *GetShapeResponse) Reset()
func (*GetShapeResponse) String ¶
func (x *GetShapeResponse) String() string
type LoadDataRequest ¶
type LoadDataRequest struct { // Describes the path of the ColumnIO tablet to load. ColumnioTabletPath string `protobuf:"bytes,1,opt,name=columnio_tablet_path,json=columnioTabletPath,proto3" json:"columnio_tablet_path,omitempty"` // Describes the field to load within the ColumnIO tablet. ColumnioField string `protobuf:"bytes,2,opt,name=columnio_field,json=columnioField,proto3" json:"columnio_field,omitempty"` // Individual element shape, excluding rows. ElementShape *data.ShapeProto `protobuf:"bytes,3,opt,name=element_shape,json=elementShape,proto3" json:"element_shape,omitempty"` // Warning: ColumnIO does not support random-access, so use offset with // caution in performance-critical scenarios. Offset int64 `protobuf:"varint,4,opt,name=offset,proto3" json:"offset,omitempty"` // Maximum number of elements (with shape element_shape) to load. Limit int64 `protobuf:"varint,5,opt,name=limit,proto3" json:"limit,omitempty"` // If more than one item is requested (via limit > 1), then this request // attribute zips together the produced vectors. Zip bool `protobuf:"varint,6,opt,name=zip,proto3" json:"zip,omitempty"` // contains filtered or unexported fields }
func (*LoadDataRequest) Descriptor
deprecated
func (*LoadDataRequest) Descriptor() ([]byte, []int)
Deprecated: Use LoadDataRequest.ProtoReflect.Descriptor instead.
func (*LoadDataRequest) GetColumnioField ¶
func (x *LoadDataRequest) GetColumnioField() string
func (*LoadDataRequest) GetColumnioTabletPath ¶
func (x *LoadDataRequest) GetColumnioTabletPath() string
func (*LoadDataRequest) GetElementShape ¶
func (x *LoadDataRequest) GetElementShape() *data.ShapeProto
func (*LoadDataRequest) GetLimit ¶
func (x *LoadDataRequest) GetLimit() int64
func (*LoadDataRequest) GetOffset ¶
func (x *LoadDataRequest) GetOffset() int64
func (*LoadDataRequest) GetZip ¶
func (x *LoadDataRequest) GetZip() bool
func (*LoadDataRequest) ProtoMessage ¶
func (*LoadDataRequest) ProtoMessage()
func (*LoadDataRequest) ProtoReflect ¶
func (x *LoadDataRequest) ProtoReflect() protoreflect.Message
func (*LoadDataRequest) Reset ¶
func (x *LoadDataRequest) Reset()
func (*LoadDataRequest) String ¶
func (x *LoadDataRequest) String() string
type LoadDataResponse ¶
type LoadDataResponse struct { Data *data.GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"` DataShape *data.ShapeProto `protobuf:"bytes,2,opt,name=data_shape,json=dataShape,proto3" json:"data_shape,omitempty"` AvailableRows int64 `protobuf:"varint,3,opt,name=available_rows,json=availableRows,proto3" json:"available_rows,omitempty"` RowsLoaded int64 `protobuf:"varint,4,opt,name=rows_loaded,json=rowsLoaded,proto3" json:"rows_loaded,omitempty"` Nanoseconds int64 `protobuf:"varint,5,opt,name=nanoseconds,proto3" json:"nanoseconds,omitempty"` // contains filtered or unexported fields }
func (*LoadDataResponse) Descriptor
deprecated
func (*LoadDataResponse) Descriptor() ([]byte, []int)
Deprecated: Use LoadDataResponse.ProtoReflect.Descriptor instead.
func (*LoadDataResponse) GetAvailableRows ¶
func (x *LoadDataResponse) GetAvailableRows() int64
func (*LoadDataResponse) GetData ¶
func (x *LoadDataResponse) GetData() *data.GlobalDataHandle
func (*LoadDataResponse) GetDataShape ¶
func (x *LoadDataResponse) GetDataShape() *data.ShapeProto
func (*LoadDataResponse) GetNanoseconds ¶
func (x *LoadDataResponse) GetNanoseconds() int64
func (*LoadDataResponse) GetRowsLoaded ¶
func (x *LoadDataResponse) GetRowsLoaded() int64
func (*LoadDataResponse) ProtoMessage ¶
func (*LoadDataResponse) ProtoMessage()
func (*LoadDataResponse) ProtoReflect ¶
func (x *LoadDataResponse) ProtoReflect() protoreflect.Message
func (*LoadDataResponse) Reset ¶
func (x *LoadDataResponse) Reset()
func (*LoadDataResponse) String ¶
func (x *LoadDataResponse) String() string
type ResetDeviceRequest ¶
type ResetDeviceRequest struct { DeviceHandle *data.DeviceHandle `protobuf:"bytes,1,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"` // contains filtered or unexported fields }
func (*ResetDeviceRequest) Descriptor
deprecated
func (*ResetDeviceRequest) Descriptor() ([]byte, []int)
Deprecated: Use ResetDeviceRequest.ProtoReflect.Descriptor instead.
func (*ResetDeviceRequest) GetDeviceHandle ¶
func (x *ResetDeviceRequest) GetDeviceHandle() *data.DeviceHandle
func (*ResetDeviceRequest) ProtoMessage ¶
func (*ResetDeviceRequest) ProtoMessage()
func (*ResetDeviceRequest) ProtoReflect ¶
func (x *ResetDeviceRequest) ProtoReflect() protoreflect.Message
func (*ResetDeviceRequest) Reset ¶
func (x *ResetDeviceRequest) Reset()
func (*ResetDeviceRequest) String ¶
func (x *ResetDeviceRequest) String() string
type ResetDeviceResponse ¶
type ResetDeviceResponse struct {
// contains filtered or unexported fields
}
func (*ResetDeviceResponse) Descriptor
deprecated
func (*ResetDeviceResponse) Descriptor() ([]byte, []int)
Deprecated: Use ResetDeviceResponse.ProtoReflect.Descriptor instead.
func (*ResetDeviceResponse) ProtoMessage ¶
func (*ResetDeviceResponse) ProtoMessage()
func (*ResetDeviceResponse) ProtoReflect ¶
func (x *ResetDeviceResponse) ProtoReflect() protoreflect.Message
func (*ResetDeviceResponse) Reset ¶
func (x *ResetDeviceResponse) Reset()
func (*ResetDeviceResponse) String ¶
func (x *ResetDeviceResponse) String() string
type TransferFromOutfeedRequest ¶
type TransferFromOutfeedRequest struct { // This optional field directs the service to return the literal in this // layout. A shape is used to hold the layout to accommodate tuples. ShapeWithLayout *data.ShapeProto `protobuf:"bytes,1,opt,name=shape_with_layout,json=shapeWithLayout,proto3" json:"shape_with_layout,omitempty"` ReplicaId int64 `protobuf:"varint,2,opt,name=replica_id,json=replicaId,proto3" json:"replica_id,omitempty"` DeviceHandle *data.DeviceHandle `protobuf:"bytes,3,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"` // contains filtered or unexported fields }
func (*TransferFromOutfeedRequest) Descriptor
deprecated
func (*TransferFromOutfeedRequest) Descriptor() ([]byte, []int)
Deprecated: Use TransferFromOutfeedRequest.ProtoReflect.Descriptor instead.
func (*TransferFromOutfeedRequest) GetDeviceHandle ¶
func (x *TransferFromOutfeedRequest) GetDeviceHandle() *data.DeviceHandle
func (*TransferFromOutfeedRequest) GetReplicaId ¶
func (x *TransferFromOutfeedRequest) GetReplicaId() int64
func (*TransferFromOutfeedRequest) GetShapeWithLayout ¶
func (x *TransferFromOutfeedRequest) GetShapeWithLayout() *data.ShapeProto
func (*TransferFromOutfeedRequest) ProtoMessage ¶
func (*TransferFromOutfeedRequest) ProtoMessage()
func (*TransferFromOutfeedRequest) ProtoReflect ¶
func (x *TransferFromOutfeedRequest) ProtoReflect() protoreflect.Message
func (*TransferFromOutfeedRequest) Reset ¶
func (x *TransferFromOutfeedRequest) Reset()
func (*TransferFromOutfeedRequest) String ¶
func (x *TransferFromOutfeedRequest) String() string
type TransferFromOutfeedResponse ¶
type TransferFromOutfeedResponse struct { Literal *data.LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"` // contains filtered or unexported fields }
func (*TransferFromOutfeedResponse) Descriptor
deprecated
func (*TransferFromOutfeedResponse) Descriptor() ([]byte, []int)
Deprecated: Use TransferFromOutfeedResponse.ProtoReflect.Descriptor instead.
func (*TransferFromOutfeedResponse) GetLiteral ¶
func (x *TransferFromOutfeedResponse) GetLiteral() *data.LiteralProto
func (*TransferFromOutfeedResponse) ProtoMessage ¶
func (*TransferFromOutfeedResponse) ProtoMessage()
func (*TransferFromOutfeedResponse) ProtoReflect ¶
func (x *TransferFromOutfeedResponse) ProtoReflect() protoreflect.Message
func (*TransferFromOutfeedResponse) Reset ¶
func (x *TransferFromOutfeedResponse) Reset()
func (*TransferFromOutfeedResponse) String ¶
func (x *TransferFromOutfeedResponse) String() string
type TransferToClientRequest ¶
type TransferToClientRequest struct { Data *data.GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"` // This optional field directs the service to return the literal in this // layout. A shape is used to hold the layout to accommodate tuples. ShapeWithLayout *data.ShapeProto `protobuf:"bytes,2,opt,name=shape_with_layout,json=shapeWithLayout,proto3" json:"shape_with_layout,omitempty"` // contains filtered or unexported fields }
func (*TransferToClientRequest) Descriptor
deprecated
func (*TransferToClientRequest) Descriptor() ([]byte, []int)
Deprecated: Use TransferToClientRequest.ProtoReflect.Descriptor instead.
func (*TransferToClientRequest) GetData ¶
func (x *TransferToClientRequest) GetData() *data.GlobalDataHandle
func (*TransferToClientRequest) GetShapeWithLayout ¶
func (x *TransferToClientRequest) GetShapeWithLayout() *data.ShapeProto
func (*TransferToClientRequest) ProtoMessage ¶
func (*TransferToClientRequest) ProtoMessage()
func (*TransferToClientRequest) ProtoReflect ¶
func (x *TransferToClientRequest) ProtoReflect() protoreflect.Message
func (*TransferToClientRequest) Reset ¶
func (x *TransferToClientRequest) Reset()
func (*TransferToClientRequest) String ¶
func (x *TransferToClientRequest) String() string
type TransferToClientResponse ¶
type TransferToClientResponse struct { Literal *data.LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"` // contains filtered or unexported fields }
func (*TransferToClientResponse) Descriptor
deprecated
func (*TransferToClientResponse) Descriptor() ([]byte, []int)
Deprecated: Use TransferToClientResponse.ProtoReflect.Descriptor instead.
func (*TransferToClientResponse) GetLiteral ¶
func (x *TransferToClientResponse) GetLiteral() *data.LiteralProto
func (*TransferToClientResponse) ProtoMessage ¶
func (*TransferToClientResponse) ProtoMessage()
func (*TransferToClientResponse) ProtoReflect ¶
func (x *TransferToClientResponse) ProtoReflect() protoreflect.Message
func (*TransferToClientResponse) Reset ¶
func (x *TransferToClientResponse) Reset()
func (*TransferToClientResponse) String ¶
func (x *TransferToClientResponse) String() string
type TransferToInfeedRequest ¶
type TransferToInfeedRequest struct { Literal *data.LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"` ReplicaId int64 `protobuf:"varint,2,opt,name=replica_id,json=replicaId,proto3" json:"replica_id,omitempty"` DeviceHandle *data.DeviceHandle `protobuf:"bytes,3,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"` // contains filtered or unexported fields }
func (*TransferToInfeedRequest) Descriptor
deprecated
func (*TransferToInfeedRequest) Descriptor() ([]byte, []int)
Deprecated: Use TransferToInfeedRequest.ProtoReflect.Descriptor instead.
func (*TransferToInfeedRequest) GetDeviceHandle ¶
func (x *TransferToInfeedRequest) GetDeviceHandle() *data.DeviceHandle
func (*TransferToInfeedRequest) GetLiteral ¶
func (x *TransferToInfeedRequest) GetLiteral() *data.LiteralProto
func (*TransferToInfeedRequest) GetReplicaId ¶
func (x *TransferToInfeedRequest) GetReplicaId() int64
func (*TransferToInfeedRequest) ProtoMessage ¶
func (*TransferToInfeedRequest) ProtoMessage()
func (*TransferToInfeedRequest) ProtoReflect ¶
func (x *TransferToInfeedRequest) ProtoReflect() protoreflect.Message
func (*TransferToInfeedRequest) Reset ¶
func (x *TransferToInfeedRequest) Reset()
func (*TransferToInfeedRequest) String ¶
func (x *TransferToInfeedRequest) String() string
type TransferToInfeedResponse ¶
type TransferToInfeedResponse struct {
// contains filtered or unexported fields
}
func (*TransferToInfeedResponse) Descriptor
deprecated
func (*TransferToInfeedResponse) Descriptor() ([]byte, []int)
Deprecated: Use TransferToInfeedResponse.ProtoReflect.Descriptor instead.
func (*TransferToInfeedResponse) ProtoMessage ¶
func (*TransferToInfeedResponse) ProtoMessage()
func (*TransferToInfeedResponse) ProtoReflect ¶
func (x *TransferToInfeedResponse) ProtoReflect() protoreflect.Message
func (*TransferToInfeedResponse) Reset ¶
func (x *TransferToInfeedResponse) Reset()
func (*TransferToInfeedResponse) String ¶
func (x *TransferToInfeedResponse) String() string
type TransferToServerRequest ¶
type TransferToServerRequest struct { Literal *data.LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"` DeviceHandle *data.DeviceHandle `protobuf:"bytes,2,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"` // contains filtered or unexported fields }
func (*TransferToServerRequest) Descriptor
deprecated
func (*TransferToServerRequest) Descriptor() ([]byte, []int)
Deprecated: Use TransferToServerRequest.ProtoReflect.Descriptor instead.
func (*TransferToServerRequest) GetDeviceHandle ¶
func (x *TransferToServerRequest) GetDeviceHandle() *data.DeviceHandle
func (*TransferToServerRequest) GetLiteral ¶
func (x *TransferToServerRequest) GetLiteral() *data.LiteralProto
func (*TransferToServerRequest) ProtoMessage ¶
func (*TransferToServerRequest) ProtoMessage()
func (*TransferToServerRequest) ProtoReflect ¶
func (x *TransferToServerRequest) ProtoReflect() protoreflect.Message
func (*TransferToServerRequest) Reset ¶
func (x *TransferToServerRequest) Reset()
func (*TransferToServerRequest) String ¶
func (x *TransferToServerRequest) String() string
type TransferToServerResponse ¶
type TransferToServerResponse struct { Data *data.GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"` // contains filtered or unexported fields }
func (*TransferToServerResponse) Descriptor
deprecated
func (*TransferToServerResponse) Descriptor() ([]byte, []int)
Deprecated: Use TransferToServerResponse.ProtoReflect.Descriptor instead.
func (*TransferToServerResponse) GetData ¶
func (x *TransferToServerResponse) GetData() *data.GlobalDataHandle
func (*TransferToServerResponse) ProtoMessage ¶
func (*TransferToServerResponse) ProtoMessage()
func (*TransferToServerResponse) ProtoReflect ¶
func (x *TransferToServerResponse) ProtoReflect() protoreflect.Message
func (*TransferToServerResponse) Reset ¶
func (x *TransferToServerResponse) Reset()
func (*TransferToServerResponse) String ¶
func (x *TransferToServerResponse) String() string
type UnpackRequest ¶
type UnpackRequest struct { Data *data.GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"` // contains filtered or unexported fields }
func (*UnpackRequest) Descriptor
deprecated
func (*UnpackRequest) Descriptor() ([]byte, []int)
Deprecated: Use UnpackRequest.ProtoReflect.Descriptor instead.
func (*UnpackRequest) GetData ¶
func (x *UnpackRequest) GetData() *data.GlobalDataHandle
func (*UnpackRequest) ProtoMessage ¶
func (*UnpackRequest) ProtoMessage()
func (*UnpackRequest) ProtoReflect ¶
func (x *UnpackRequest) ProtoReflect() protoreflect.Message
func (*UnpackRequest) Reset ¶
func (x *UnpackRequest) Reset()
func (*UnpackRequest) String ¶
func (x *UnpackRequest) String() string
type UnpackResponse ¶
type UnpackResponse struct { TiedData []*data.GlobalDataHandle `protobuf:"bytes,1,rep,name=tied_data,json=tiedData,proto3" json:"tied_data,omitempty"` // contains filtered or unexported fields }
func (*UnpackResponse) Descriptor
deprecated
func (*UnpackResponse) Descriptor() ([]byte, []int)
Deprecated: Use UnpackResponse.ProtoReflect.Descriptor instead.
func (*UnpackResponse) GetTiedData ¶
func (x *UnpackResponse) GetTiedData() []*data.GlobalDataHandle
func (*UnpackResponse) ProtoMessage ¶
func (*UnpackResponse) ProtoMessage()
func (*UnpackResponse) ProtoReflect ¶
func (x *UnpackResponse) ProtoReflect() protoreflect.Message
func (*UnpackResponse) Reset ¶
func (x *UnpackResponse) Reset()
func (*UnpackResponse) String ¶
func (x *UnpackResponse) String() string
type UnregisterRequest ¶
type UnregisterRequest struct { Data []*data.GlobalDataHandle `protobuf:"bytes,1,rep,name=data,proto3" json:"data,omitempty"` // contains filtered or unexported fields }
func (*UnregisterRequest) Descriptor
deprecated
func (*UnregisterRequest) Descriptor() ([]byte, []int)
Deprecated: Use UnregisterRequest.ProtoReflect.Descriptor instead.
func (*UnregisterRequest) GetData ¶
func (x *UnregisterRequest) GetData() []*data.GlobalDataHandle
func (*UnregisterRequest) ProtoMessage ¶
func (*UnregisterRequest) ProtoMessage()
func (*UnregisterRequest) ProtoReflect ¶
func (x *UnregisterRequest) ProtoReflect() protoreflect.Message
func (*UnregisterRequest) Reset ¶
func (x *UnregisterRequest) Reset()
func (*UnregisterRequest) String ¶
func (x *UnregisterRequest) String() string
type UnregisterResponse ¶
type UnregisterResponse struct {
// contains filtered or unexported fields
}
func (*UnregisterResponse) Descriptor
deprecated
func (*UnregisterResponse) Descriptor() ([]byte, []int)
Deprecated: Use UnregisterResponse.ProtoReflect.Descriptor instead.
func (*UnregisterResponse) ProtoMessage ¶
func (*UnregisterResponse) ProtoMessage()
func (*UnregisterResponse) ProtoReflect ¶
func (x *UnregisterResponse) ProtoReflect() protoreflect.Message
func (*UnregisterResponse) Reset ¶
func (x *UnregisterResponse) Reset()
func (*UnregisterResponse) String ¶
func (x *UnregisterResponse) String() string
type WaitForExecutionRequest ¶
type WaitForExecutionRequest struct { Execution *data.ExecutionHandle `protobuf:"bytes,1,opt,name=execution,proto3" json:"execution,omitempty"` // contains filtered or unexported fields }
func (*WaitForExecutionRequest) Descriptor
deprecated
func (*WaitForExecutionRequest) Descriptor() ([]byte, []int)
Deprecated: Use WaitForExecutionRequest.ProtoReflect.Descriptor instead.
func (*WaitForExecutionRequest) GetExecution ¶
func (x *WaitForExecutionRequest) GetExecution() *data.ExecutionHandle
func (*WaitForExecutionRequest) ProtoMessage ¶
func (*WaitForExecutionRequest) ProtoMessage()
func (*WaitForExecutionRequest) ProtoReflect ¶
func (x *WaitForExecutionRequest) ProtoReflect() protoreflect.Message
func (*WaitForExecutionRequest) Reset ¶
func (x *WaitForExecutionRequest) Reset()
func (*WaitForExecutionRequest) String ¶
func (x *WaitForExecutionRequest) String() string
type WaitForExecutionResponse ¶
type WaitForExecutionResponse struct { Output *data.GlobalDataHandle `protobuf:"bytes,1,opt,name=output,proto3" json:"output,omitempty"` Profile *data.ExecutionProfile `protobuf:"bytes,2,opt,name=profile,proto3" json:"profile,omitempty"` // contains filtered or unexported fields }
func (*WaitForExecutionResponse) Descriptor
deprecated
func (*WaitForExecutionResponse) Descriptor() ([]byte, []int)
Deprecated: Use WaitForExecutionResponse.ProtoReflect.Descriptor instead.
func (*WaitForExecutionResponse) GetOutput ¶
func (x *WaitForExecutionResponse) GetOutput() *data.GlobalDataHandle
func (*WaitForExecutionResponse) GetProfile ¶
func (x *WaitForExecutionResponse) GetProfile() *data.ExecutionProfile
func (*WaitForExecutionResponse) ProtoMessage ¶
func (*WaitForExecutionResponse) ProtoMessage()
func (*WaitForExecutionResponse) ProtoReflect ¶
func (x *WaitForExecutionResponse) ProtoReflect() protoreflect.Message
func (*WaitForExecutionResponse) Reset ¶
func (x *WaitForExecutionResponse) Reset()
func (*WaitForExecutionResponse) String ¶
func (x *WaitForExecutionResponse) String() string