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