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.