#include <cmath>
#include "tensorflow/lite/core/c/common.h"
#include "tensorflow/lite/kernels/internal/tensor_ctypes.h"
#include "tensorflow/lite/kernels/kernel_util.h"
namespace tflite {
namespace ops {
namespace builtin {
namespace atan2 {
TfLiteStatus EnsureSameShape(
TfLiteContext* context,
const TfLiteTensor* a, const TfLiteTensor* b) { … }
TfLiteStatus Atan2Prepare(TfLiteContext* context, TfLiteNode* node) { … }
template<typename Float>
TfLiteStatus Atan2(const TfLiteTensor* input_y,
const TfLiteTensor* input_x,
TfLiteTensor* output) { … }
TfLiteStatus Atan2Eval(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_ATAN2() { … }
}
}
}