TensorFlow Lite Base Operation Coverage

Tests of the TFLite's builtin operation coverage.

This focuses on lowering to tosa operations and may not have support for all IREE backends.

*Last Updated Manually: 2022/5/25

TFlite Operation Coverage

TFLite OperationStaticDynamic
Abs✓✓
Add✓✓
AddN✓✓
ArgMax✓✓
ArgMin✓✓
AssignVariable✓✓
AveragePool2d✓✓
BatchMatmul✓✓
BatchToSpaceNd✓✓
BidirectionalSequenceLstm✗✗
BidirectionalSequenceRnn✗✗
BroadcastArgs✗✗
BroadcastTo✓✗
Bucketize✗✗
Call✓✓
CallOnce✓✓
Cast✓✓
Ceil✓✓
ComplexAbs✓✓
ConcatEmbeddings✗✗
Concatenation✓✓
Conv2d✓✓
Conv3d✓✓
Conv3dTranspose✗✗
Cos✓✓
Cumsum✗✗
Custom✓✓
Delegate✓✓
Densify✗✗
DepthToSpace✓✓
DepthwiseConv2d✓✓
Dequantize✓✓
Div✓✓
DynamicUpdateSlice✗✗
Elu✓✓
EmbeddingLookup✗✗
EmbeddingLookupSparse✗✗
Equal✓✓
Exp✓✓
ExpandDims✓✓
FakeQuant✓✓
Fill✓✓
Floor✓✓
FloorDiv✓✓
FloorMod✓✓
FullyConnected✓✓
Gather✓✓
GatherNd✓✓
Gelu✗✗
Greater✓✓
GreaterEqual✓✓
HardSwish✓✓
Hashtable✗✗
HashtableFind✗✗
HashtableImport✗✗
HashtableLookup✗✗
HashtableSize✗✗
If✓✓
Imag✗✗
L2Normalization✓✓
L2Pool2d✗✗
LeakyRelu✓✓
Less✓✓
LessEqual✓✓
LocalResponseNormalization✗✗
Log✓✓
LogicalAnd✓✓
LogicalNot✓✓
LogicalOr✓✓
Logistic✓✓
LogSoftmax✓✓
LshProjection✗✗
Lstm✗✗
MatrixDiag✗✗
MatrixSetDiag✗✗
Maximum✓✓
MaxPool2d✓✓
Mean✓✓
Minimum✓✓
MirrorPad✗✗
Mul✓✓
Multinomial✗✗
Neg✓✓
NonMaxSuppressionV4✗✗
NonMaxSuppressionV5✗✗
NotEqual✓✓
OneHot✓✓
Pack✓✓
Pad✓✓
Padv2✓✓
Pow✓✓
Prelu✓✓
Quantize✓✓
RandomStandardNormal✗✗
RandomUniform✗✗
Range✓✗
Rank✓✓
ReadVariable✓✓
Real✗✗
ReduceAll✓✓
ReduceAny✓✓
ReduceMax✓✓
ReduceMin✓✓
ReduceProd✓✓
Relu✓✓
Relu6✓✓
ReluN1To1✗✗
Reshape✓✓
ResizeBilinear✓✓
ResizeNearestNeighbor✓✓
ReverseSequence✗✗
ReverseV2✓✓
Rfft2d✗✗
Rnn✗✗
Round✓✓
Rsqrt✓✓
ScatterNd✗✗
SegmentSum✗✗
Select✓✓
SelectV2✓✓
Shape✓✓
Sin✓✓
SkipGram✗✗
Slice✓✓
Softmax✓✓
SpaceToBatchNd✓✓
SpaceToDepth✓✓
SparseToDense✓✗
Split✓✓
SplitV✓✓
Sqrt✓✓
Square✓✓
SquaredDifference✓✓
Squeeze✓✓
StridedSlice✓✓
Sub✓✓
Sum✓✓
Svdf✗✗
Tanh✓✓
Tile✓✓
TopkV2✗✗
Transpose✓✓
TransposeConv✓✗
UnidirectionalSequenceLstm✗✗
UnidirectionalSequenceRnn✗✗
Unique✗✗
Unpack✓✓
VarHandle✓✓
Where✗✗
While✓✓
ZerosLike✓✗