#include <stdint.h>
#include "tensorflow/lite/core/c/builtin_op_data.h"
#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 one_hot {
constexpr int kIndicesTensor = …;
constexpr int kDepthTensor = …;
constexpr int kOnValueTensor = …;
constexpr int kOffValueTensor = …;
constexpr int kOutputTensor = …;
struct OneHotContext { … };
template <typename T, typename TI>
void OneHotComputeImpl(const OneHotContext& op_context) { … }
template <typename T>
void OneHotCompute(const OneHotContext& op_context) { … }
TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
const OneHotContext& op_context) { … }
TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_ONE_HOT() { … }
}
}
}