package xla

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

message BinaryOpRequest

xla_data.proto:599

Used in: OpRequest

enum BinaryOperation

xla_data.proto:554

Used in: BinaryOpRequest

message BroadcastRequest

xla_data.proto:454

Used in: OpRequest

message CallRequest

xla_data.proto:385

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, 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:109

message ComputationResponse

xla.proto:113

message ComputationStats

xla_data.proto:173

Statistics of a computation.

Used in: ComputationStatsResponse

message ComputationStatsRequest

xla.proto:101

message ComputationStatsResponse

xla.proto:105

message ComputeConstantRequest

xla.proto:213

message ComputeConstantResponse

xla.proto:219

message ConcatenateRequest

xla_data.proto:500

Used in: OpRequest

message ConstantRequest

xla_data.proto:312

Used in: OpRequest

message ConvertRequest

xla_data.proto:495

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:117

(message has no fields)

message CreateChannelHandleResponse

xla.proto:120

message CrossReplicaSumRequest

xla_data.proto:424

Used in: OpRequest

message CustomCallRequest

xla_data.proto:390

Used in: OpRequest

message DeconstructTupleRequest

xla.proto:223

message DeconstructTupleResponse

xla.proto:227

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:173

message ExecuteAsyncResponse

xla.proto:190

message ExecuteParallelRequest

xla.proto:160

message ExecuteParallelResponse

xla.proto:169

message ExecuteRequest

xla.proto:139

Used in: ExecuteParallelRequest

message ExecuteResponse

xla.proto:164

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 ExecutionProfile

xla_data.proto:182

Profile data from the execution of a computation.

Used in: ExecuteResponse, WaitForExecutionResponse

message GetComputationShapeRequest

xla.proto:277

message GetComputationShapeResponse

xla.proto:281

message GetDeviceHandlesRequest

xla.proto:39

message GetDeviceHandlesResponse

xla.proto:43

message GetLocalShapeRequest

xla_data.proto:481

message GetLocalShapeResponse

xla_data.proto:486

message GetShapeRequest

xla.proto:269

message GetShapeResponse

xla.proto:273

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:204

message IsConstantResponse

xla.proto:209

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:31

message LoadComputationSnapshotResponse

xla.proto:35

message LoadDataRequest

xla.proto:231

message LoadDataResponse

xla.proto:253

message MapRequest

xla_data.proto:396

Used in: OpRequest

message OpRequest

xla_data.proto:674

Used in: OperationRequest

message OpResponse

xla_data.proto:712

message OperationRequest

session.proto:26

Describes a single operation request.

Used in: SessionComputation

message PadRequest

xla_data.proto:459

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:475

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:606

Used in: RngRequest

message RecvRequest

xla_data.proto:669

Used in: OpRequest

message ReduceRequest

xla_data.proto:402

Used in: OpRequest

message ReduceWindowRequest

xla_data.proto:417

Used in: OpRequest

message ResetDeviceRequest

xla.proto:94

message ResetDeviceResponse

xla.proto:98

(message has no fields)

message ReshapeRequest

xla_data.proto:465

Used in: OpRequest

message ReverseRequest

xla_data.proto:449

Used in: OpRequest

message RngRequest

xla_data.proto:622

Used in: OpRequest

message SelectAndScatterRequest

xla_data.proto:428

Used in: OpRequest

message SendRequest

xla_data.proto:664

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:74

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:131

message SetReturnValueResponse

xla.proto:136

(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, ExecuteAsyncRequest, ExecuteRequest, 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:23

message SnapshotComputationResponse

xla.proto:27

message SpecializeRequest

xla.proto:261

message SpecializeResponse

xla.proto:266

(message has no fields)

message TernaryOpRequest

xla_data.proto:645

Used in: OpRequest

enum TernaryOperation

xla_data.proto:628

Used in: TernaryOpRequest

message TraceRequest

xla_data.proto:490

Used in: OpRequest, SessionComputation

message TransferToClientInProcessRequest

xla.proto:77

message TransferToClientInProcessResponse

xla.proto:82

(message has no fields)

message TransferToClientRequest

xla.proto:47

message TransferToClientResponse

xla.proto:55

message TransferToInfeedRequest

xla.proto:85

message TransferToInfeedResponse

xla.proto:91

(message has no fields)

message TransferToServerInProcessRequest

xla.proto:68

message TransferToServerInProcessResponse

xla.proto:73

message TransferToServerRequest

xla.proto:59

message TransferToServerResponse

xla.proto:64

message UnaryOpRequest

xla_data.proto:549

Used in: OpRequest

enum UnaryOperation

xla_data.proto:515

Used in: UnaryOpRequest

message UnpackRequest

xla.proto:285

message UnpackResponse

xla.proto:289

message UnregisterRequest

xla.proto:124

message UnregisterResponse

xla.proto:128

(message has no fields)

message VariadicOpRequest

xla_data.proto:659

Used in: OpRequest

enum VariadicOperation

xla_data.proto:652

Used in: VariadicOpRequest

message WaitForExecutionRequest

xla.proto:195

message WaitForExecutionResponse

xla.proto:199

message WhileRequest

xla_data.proto:509

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