#include <complex>
#include "tensorflow/lite/core/c/common.h"
#include "tensorflow/lite/kernels/internal/tensor.h"
#include "tensorflow/lite/kernels/internal/tensor_ctypes.h"
#include "tensorflow/lite/kernels/kernel_util.h"
namespace tflite {
namespace ops {
namespace builtin {
namespace complex {
static const int kInputTensor = …;
static const int kOutputTensor = …;
TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { … }
template <typename T, typename ExtractF>
void ExtractData(const TfLiteTensor* input, ExtractF extract_func,
TfLiteTensor* output) { … }
TfLiteStatus EvalReal(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus EvalImag(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus EvalAbs(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_REAL() { … }
TfLiteRegistration* Register_IMAG() { … }
TfLiteRegistration* Register_COMPLEX_ABS() { … }
}
}
}