package xla

Mouse Melon logoGet desktop application:
View/edit binary Protocol Buffers messages

message BinaryOpRequest

xla_data.proto:610

Used in: OpRequest

enum BinaryOperation

xla_data.proto:565

Used in: BinaryOpRequest

message BroadcastRequest

xla_data.proto:465

Used in: OpRequest

message CallRequest

xla_data.proto:396

Used in: OpRequest

message ChannelHandle

xla_data.proto:242

Handle given to a user to represent a channel between two computations via a Send and Recv instruction pair. Channels are unbuffered, so Send Send instructions will be blocked until the data is transferred.

Used in: CreateChannelHandleResponse, RecvRequest, SendRequest

message ComputationDataHandle

xla_data.proto:228

Handle given to a user that represents a data result in a computation. This is used to pass to subsequent computations that depends upon the data as an operand.

Used in: BinaryOpRequest, BroadcastRequest, CallRequest, ComputeConstantRequest, ConcatenateRequest, ConvertRequest, ConvolveRequest, CrossReplicaSumRequest, CustomCallRequest, DynamicSliceRequest, DynamicUpdateSliceRequest, GetLocalShapeRequest, GetTupleElementRequest, IsConstantRequest, MapRequest, OpResponse, OperationRequest, OutfeedRequest, PadRequest, ReduceRequest, ReduceWindowRequest, ReshapeRequest, ReverseRequest, RngRequest, SelectAndScatterRequest, SendRequest, SetReturnValueRequest, SliceRequest, TernaryOpRequest, TraceRequest, UnaryOpRequest, VariadicOpRequest, WhileRequest

message ComputationHandle

xla_data.proto:208

Handle given to a user that represents a computation that the user builds up before execution.

Used in: CallRequest, ComputationResponse, ComputationStatsRequest, ComputeConstantRequest, ExecuteAsyncRequest, ExecuteRequest, GetComputationShapeRequest, GetLocalShapeRequest, IsConstantRequest, LoadComputationSnapshotResponse, MapRequest, OpRequest, ReduceRequest, ReduceWindowRequest, SelectAndScatterRequest, SessionComputation, SetReturnValueRequest, SnapshotComputationRequest, SpecializeRequest, WhileRequest

message ComputationRequest

xla.proto:137

message ComputationResponse

xla.proto:141

message ComputationStats

xla_data.proto:173

Statistics of a computation.

Used in: ComputationStatsResponse

message ComputationStatsRequest

xla.proto:129

message ComputationStatsResponse

xla.proto:133

message ComputeConstantRequest

xla.proto:227

message ComputeConstantResponse

xla.proto:233

message ConcatenateRequest

xla_data.proto:511

Used in: OpRequest

message ConstantRequest

xla_data.proto:312

Used in: OpRequest

message ConvertRequest

xla_data.proto:506

Used in: OpRequest

message ConvolutionDimensionNumbers

xla_data.proto:346

Used in: ConvolveRequest

message ConvolveRequest

xla_data.proto:373

Used in: OpRequest

message CreateChannelHandleRequest

xla.proto:145

(message has no fields)

message CreateChannelHandleResponse

xla.proto:148

message CrossReplicaSumRequest

xla_data.proto:435

Used in: OpRequest

message CustomCallRequest

xla_data.proto:401

Used in: OpRequest

message DeconstructTupleRequest

xla.proto:237

message DeconstructTupleResponse

xla.proto:241

message DeviceHandle

xla_data.proto:235

Handle given to a user that represents a device to execute a computation. When replication is enabled, the device handle represents the device for the replica id 0.

Used in: ExecuteRequest, GetDeviceHandlesResponse, ResetDeviceRequest, TransferToInfeedRequest, TransferToServerRequest

message DynamicSliceRequest

xla_data.proto:327

Used in: OpRequest

message DynamicUpdateSliceRequest

xla_data.proto:337

Used in: OpRequest

message ExecuteAsyncRequest

xla.proto:194

message ExecuteAsyncResponse

xla.proto:204

message ExecuteParallelRequest

xla.proto:181

message ExecuteParallelResponse

xla.proto:190

message ExecuteRequest

xla.proto:167

Used in: ExecuteParallelRequest

message ExecuteResponse

xla.proto:185

Used in: ExecuteParallelResponse

message ExecutionHandle

xla_data.proto:214

Handle given to a user that represents an execution that the user launched asynchronously on the device.

Used in: ExecuteAsyncResponse, WaitForExecutionRequest

message ExecutionOptions

xla.proto:27

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.

Used in: ExecuteAsyncRequest, ExecuteRequest

message ExecutionProfile

xla_data.proto:182

Profile data from the execution of a computation.

Used in: ExecuteResponse, WaitForExecutionResponse

message GetComputationShapeRequest

xla.proto:291

message GetComputationShapeResponse

xla.proto:295

message GetDeviceHandlesRequest

xla.proto:67

message GetDeviceHandlesResponse

xla.proto:71

message GetLocalShapeRequest

xla_data.proto:492

message GetLocalShapeResponse

xla_data.proto:497

message GetShapeRequest

xla.proto:283

message GetShapeResponse

xla.proto:287

message GetTupleElementRequest

xla_data.proto:316

Used in: OpRequest

message GlobalDataHandle

xla_data.proto:221

Handle given to a user that represents a globally accessible allocation. Contrast this against a ComputationDataHandle, which is not globally accessible, since it only exists within a specific computation.

Used in: ComputeConstantResponse, DeconstructTupleRequest, DeconstructTupleResponse, ExecuteAsyncRequest, ExecuteRequest, ExecuteResponse, GetShapeRequest, LoadDataResponse, SpecializeRequest, TransferToClientInProcessRequest, TransferToClientRequest, TransferToServerInProcessResponse, TransferToServerResponse, UnpackRequest, UnpackResponse, UnregisterRequest, WaitForExecutionResponse

message InfeedRequest

xla_data.proto:380

Used in: OpRequest

message IsConstantRequest

xla.proto:218

message IsConstantResponse

xla.proto:223

message Layout

xla_data.proto:113

A layout describes how the array is placed in (1D) memory space. This includes the minor-to-major ordering of dimensions within a shape, as well as any padding present in those dimensions. Clients must specify the layouts of input Literals to the computation. Layouts specified in interior operations which take Shapes (for example, Convert) are ignored. See the XLA documentation for more information on shapes and layouts.

Used in: ComputeConstantRequest, Shape

message Literal

xla_data.proto:252

Literals are used when the server and client need to exchange materialized data / results. Literals are also used to describe constants used in computations. Transfers to/from the client are encoded in literal form, and the structure of the repeated fields is implied by the shape.

Used in: ConstantRequest, SessionModule, TransferToClientResponse, TransferToInfeedRequest, TransferToServerRequest

message LoadComputationSnapshotRequest

xla.proto:59

message LoadComputationSnapshotResponse

xla.proto:63

message LoadDataRequest

xla.proto:245

message LoadDataResponse

xla.proto:267

message MapRequest

xla_data.proto:407

Used in: OpRequest

message OpRequest

xla_data.proto:685

Used in: OperationRequest

message OpResponse

xla_data.proto:724

message OperationRequest

session.proto:26

Describes a single operation request.

Used in: SessionComputation

message OutfeedRequest

xla_data.proto:388

Used in: OpRequest, SessionComputation

message PadRequest

xla_data.proto:470

Used in: OpRequest

message PaddingConfig

xla_data.proto:87

Describes the padding configuration for Pad operation. The padding amount on both edges as well as between the elements are specified for each dimension.

Used in: PadRequest

message PaddingConfig.PaddingConfigDimension

xla_data.proto:89

Describes the padding configuration for a dimension.

Used in: PaddingConfig

enum PaddingValue

xla_data.proto:64

Describes the value held inside padding elements.

Used in: Layout

message ParameterRequest

xla_data.proto:486

Used in: OpRequest

enum PrimitiveType

xla_data.proto:23

Primitive types are the individual values that can be held in rectangular multidimensional arrays. A description of the rectangular multidimensional array dimensions / primitive type is given by Shape, below.

Used in: ConvertRequest, Shape

message ProgramShape

xla_data.proto:166

Shape of the parameters and output of a computation (like a traditional function signature).

Used in: GetComputationShapeResponse

enum RandomDistribution

xla_data.proto:617

Used in: RngRequest

message RecvRequest

xla_data.proto:680

Used in: OpRequest

message ReduceRequest

xla_data.proto:413

Used in: OpRequest

message ReduceWindowRequest

xla_data.proto:428

Used in: OpRequest

message ResetDeviceRequest

xla.proto:122

message ResetDeviceResponse

xla.proto:126

(message has no fields)

message ReshapeRequest

xla_data.proto:476

Used in: OpRequest

message ReverseRequest

xla_data.proto:460

Used in: OpRequest

message RngRequest

xla_data.proto:633

Used in: OpRequest

message SelectAndScatterRequest

xla_data.proto:439

Used in: OpRequest

message SendRequest

xla_data.proto:675

Used in: OpRequest, SessionComputation

message SessionComputation

session.proto:50

Describes a sequence of operation requests which define an XLA computation.

Used in: SessionModule

message SessionModule

session.proto:77

Describes a group of SessionComputations with an "entry point" computation that may refer to the other non-entry (AKA embedded) computations. This message is used to serialize a computation that has been built via the XLA service API, along with its dependencies, for purposes such as analysis/replay/file-storage.

Used in: LoadComputationSnapshotRequest, SnapshotComputationResponse

message SetReturnValueRequest

xla.proto:159

message SetReturnValueResponse

xla.proto:164

(message has no fields)

message Shape

xla_data.proto:140

A shape describes the number of dimensions in the array, the size of each dimension, and the primitive component type. Tuples are a special case in that they have rank zero and have tuple_shapes defined. See the XLA documentation for more information on shapes and layouts.

Used in: CustomCallRequest, ExecutionOptions, GetLocalShapeResponse, GetShapeResponse, InfeedRequest, Literal, LoadDataRequest, LoadDataResponse, OperationRequest, ParameterRequest, ProgramShape, RecvRequest, RngRequest, TransferToClientRequest, TransferToServerInProcessRequest

message SliceRequest

xla_data.proto:321

Used in: OpRequest

message SnapshotComputationRequest

xla.proto:51

message SnapshotComputationResponse

xla.proto:55

message SpecializeRequest

xla.proto:275

message SpecializeResponse

xla.proto:280

(message has no fields)

message TernaryOpRequest

xla_data.proto:656

Used in: OpRequest

enum TernaryOperation

xla_data.proto:639

Used in: TernaryOpRequest

message TraceRequest

xla_data.proto:501

Used in: OpRequest, SessionComputation

message TransferToClientInProcessRequest

xla.proto:105

message TransferToClientInProcessResponse

xla.proto:110

(message has no fields)

message TransferToClientRequest

xla.proto:75

message TransferToClientResponse

xla.proto:83

message TransferToInfeedRequest

xla.proto:113

message TransferToInfeedResponse

xla.proto:119

(message has no fields)

message TransferToServerInProcessRequest

xla.proto:96

message TransferToServerInProcessResponse

xla.proto:101

message TransferToServerRequest

xla.proto:87

message TransferToServerResponse

xla.proto:92

message UnaryOpRequest

xla_data.proto:560

Used in: OpRequest

enum UnaryOperation

xla_data.proto:526

Used in: UnaryOpRequest

message UnpackRequest

xla.proto:299

message UnpackResponse

xla.proto:303

message UnregisterRequest

xla.proto:152

message UnregisterResponse

xla.proto:156

(message has no fields)

message VariadicOpRequest

xla_data.proto:670

Used in: OpRequest

enum VariadicOperation

xla_data.proto:663

Used in: VariadicOpRequest

message WaitForExecutionRequest

xla.proto:209

message WaitForExecutionResponse

xla.proto:213

message WhileRequest

xla_data.proto:520

Used in: OpRequest

message Window

xla_data.proto:305

Describes the windowing in an operation such as convolution. The window is moved across a base area and for each position of the window a computation is performed. The field below describes the window and the movement of the window across a base area.

Used in: ConvolveRequest, ReduceWindowRequest, SelectAndScatterRequest

message WindowDimension

xla_data.proto:265

Used in: Window