xla

package
v0.0.0-...-3233e8b Latest Latest
Warning

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

 Go to latest Published: Dec 8, 2023 License: BSD-3-Clause Imports: 6 Imported by: 8

Documentation

Index

Constants

This section is empty.

Variables

View Source 
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.

View Source 
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.

View Source 
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 */
	// Enables shared constants for XLA/GPU. This allows large constants to be
	// 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

Source Files

View all Source files

Directories

Path Synopsis
pjrt
distributed
python
tpu_driver
gpu

Jump to

Keyboard shortcuts

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