mirror of
https://github.com/vee1e/tflite-micro.git
synced 2026-09-02 02:07:27 +00:00
f22888af65
Second OP for the TFLM Signal library, Real-Valued Fast Fourier Transform. The RFFT OP provides three resolutions: `FLOAT, INT16, INT32` Similar usage as to previous Window OP: * `op_resolver.AddRfft()` (which will add all resolutions, and determine the type at runtime) * `op_resolver.AddRfftFloat()`, `op_resolver.AddRfftInt16()`, `op_resolver.AddRfftInt32()` for a specific resolution type. * or via python as can be seen in `fft_ops_test.py` 3 testing options are provided: * Micro(C++): bazel run signal/micro/kernels:fft_test * Tensorflow/Micro(Python): bazel run python/tflite_micro/signal:fft_ops_test * Makefile(C++): make -f tensorflow/lite/micro/tools/make/Makefile test_kernel_fft_test BUG=[287346710](http://b/287346710)
125 lines
2.5 KiB
C++
125 lines
2.5 KiB
C++
/* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
==============================================================================*/
|
|
|
|
#include "python/tflite_micro/python_ops_resolver.h"
|
|
|
|
#include "tensorflow/lite/micro/kernels/micro_ops.h"
|
|
|
|
namespace tflite {
|
|
|
|
PythonOpsResolver::PythonOpsResolver() {
|
|
// Please keep this list of Builtin Operators in alphabetical order.
|
|
AddAbs();
|
|
AddAdd();
|
|
AddAddN();
|
|
AddArgMax();
|
|
AddArgMin();
|
|
AddAssignVariable();
|
|
AddAveragePool2D();
|
|
AddBatchToSpaceNd();
|
|
AddBroadcastArgs();
|
|
AddBroadcastTo();
|
|
AddCallOnce();
|
|
AddCast();
|
|
AddCeil();
|
|
AddCircularBuffer();
|
|
AddConcatenation();
|
|
AddConv2D();
|
|
AddCos();
|
|
AddCumSum();
|
|
AddDepthToSpace();
|
|
AddDepthwiseConv2D();
|
|
AddDequantize();
|
|
AddDetectionPostprocess();
|
|
AddDiv();
|
|
AddElu();
|
|
AddEqual();
|
|
AddEthosU();
|
|
AddExp();
|
|
AddExpandDims();
|
|
AddFill();
|
|
AddFloor();
|
|
AddFloorDiv();
|
|
AddFloorMod();
|
|
AddFullyConnected();
|
|
AddGather();
|
|
AddGatherNd();
|
|
AddGreater();
|
|
AddGreaterEqual();
|
|
AddHardSwish();
|
|
AddIf();
|
|
AddL2Normalization();
|
|
AddL2Pool2D();
|
|
AddLeakyRelu();
|
|
AddLess();
|
|
AddLessEqual();
|
|
AddLog();
|
|
AddLogicalAnd();
|
|
AddLogicalNot();
|
|
AddLogicalOr();
|
|
AddLogistic();
|
|
AddLogSoftmax();
|
|
AddMaxPool2D();
|
|
AddMaximum();
|
|
AddMean();
|
|
AddMinimum();
|
|
AddMirrorPad();
|
|
AddMul();
|
|
AddNeg();
|
|
AddNotEqual();
|
|
AddPack();
|
|
AddPad();
|
|
AddPadV2();
|
|
AddPrelu();
|
|
AddQuantize();
|
|
AddReadVariable();
|
|
AddReduceMax();
|
|
AddRelu();
|
|
AddRelu6();
|
|
AddReshape();
|
|
AddResizeBilinear();
|
|
AddResizeNearestNeighbor();
|
|
AddRfft();
|
|
AddRound();
|
|
AddRsqrt();
|
|
AddSelectV2();
|
|
AddShape();
|
|
AddSin();
|
|
AddSlice();
|
|
AddSoftmax();
|
|
AddSpaceToBatchNd();
|
|
AddSpaceToDepth();
|
|
AddSplit();
|
|
AddSplitV();
|
|
AddSqrt();
|
|
AddSquare();
|
|
AddSquaredDifference();
|
|
AddSqueeze();
|
|
AddStridedSlice();
|
|
AddSub();
|
|
AddSum();
|
|
AddSvdf();
|
|
AddTanh();
|
|
AddTranspose();
|
|
AddTransposeConv();
|
|
AddUnidirectionalSequenceLSTM();
|
|
AddUnpack();
|
|
AddVarHandle();
|
|
AddWhile();
|
|
AddWindow();
|
|
AddZerosLike();
|
|
}
|
|
|
|
} // namespace tflite
|