TensorFlow Neuron (tensorflow-neuron (TF1.x)) Supported operators [XLA]
This document is relevant for: Inf1
TensorFlow Neuron (tensorflow-neuron (TF1.x)) Supported operators [XLA]#
To see a list of supported operators for XLA, run the following command:
neuron-cc list-operators --framework XLA
Supported XLA Operators |
Notes |
|---|---|
Abs |
|
Add |
|
Allgather |
|
Allreduce |
|
Atan2 |
|
Batchnorm |
|
Batchnormgrad |
|
Batchnorminference |
|
Broadcast |
|
BroadcastInDim |
|
Ceil |
|
Clamp |
|
Compare |
|
Concatenate |
|
Constant |
|
ConstantLiteral |
|
ConvertElementType |
|
Cos |
|
Customcall |
|
Div |
|
Dot |
|
DotGeneral |
|
DynamicUpdateSlice |
Supports only for constant index |
Eq |
|
Exp |
|
Floor |
|
Gather |
Supports only disjoint start_index_map and remapped_offset_dims |
Ge |
|
GetTupleElement |
|
Gt |
|
Iota |
|
Le |
|
Log |
|
LogicalAnd |
|
LogicalNot |
|
Lt |
|
Max |
|
Min |
|
Mul |
|
Ne |
|
Neg |
|
Pad |
|
Pow |
Exponent argument must be a compile-time integer constant |
Reduce |
Min, Max, Add and Mul are the only supported computations. Init_values must be constant |
Reshape |
|
RngBitGenerator |
Ignores user seed |
RngUniform |
|
Rsqrt |
|
Scatter |
|
Select |
|
ShiftRightLogical |
|
Sign |
|
Sin |
|
Slice |
|
Sqrt |
|
Sub |
|
Tanh |
|
Transpose |
|
Tuple |
This document is relevant for: Inf1