org.tensorflow.op.xla

Classes

BroadcastHelper <T extends TType > Helper operator for performing XLA-style broadcasts

Broadcasts `lhs` and `rhs` to the same rank, by adding size 1 dimensions to whichever of `lhs` and `rhs` has the lower rank, using XLA's broadcasting rules for binary operators.

ClusterOutput <T extends TType > Operator that connects the output of an XLA computation to other consumer graph nodes.
Conv <T extends TType > Wraps the XLA ConvGeneralDilated operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution .

Dequantize Takes the packed uint32 input and unpacks the input to uint8 to do

Dequantization on device.

Dot <T extends TType > Wraps the XLA DotGeneral operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral .

DynamicSlice <T extends TType > Wraps the XLA DynamicSlice operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice .

DynamicUpdateSlice <T extends TType > Wraps the XLA DynamicUpdateSlice operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice .

Einsum <T extends TType > An op which supports basic einsum op with 2 inputs and 1 output.
Gather <T extends TType > Wraps the XLA Gather operator documented at

https://www.tensorflow.org/xla/operation_semantics#gather

KeyValueSort <T extends TNumber , U extends TType > Wraps the XLA Sort operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#sort .

Pad <T extends TType > Wraps the XLA Pad operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#pad .

Recv <T extends TType > Receives the named tensor from another XLA computation.
ReplicaId Replica ID.
SelfAdjointEig <T extends TType > Computes the eigen decomposition of a batch of self-adjoint matrices

(Note: Only real inputs are supported).

Send Sends the named tensor to another XLA computation.
Sharding <T extends TType > An op which shards the input based on the given sharding attribute.
Sort <T extends TType > Wraps the XLA Sort operator, documented at

https://www.tensorflow.org/performance/xla/operation_semantics#sort .

Svd <T extends TType > Computes the eigen decomposition of a batch of self-adjoint matrices

(Note: Only real inputs are supported).

XlaRecvFromHost <T extends TType > An op to receive a tensor from the host.
XlaSendToHost An op to send a tensor to the host.
XlaSetBound Set a bound for the given input value as a hint to Xla compiler,

returns the same value.