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