#include <cstddef>
#include <cstdint>
#include <memory>
#include <vector>
#include "tensorflow/lite/c/common.h"
#include "tensorflow/lite/context_util.h"
#include "tensorflow/lite/core/c/builtin_op_data.h"
#include "tensorflow/lite/core/subgraph.h"
#include "tensorflow/lite/kernels/control_flow_common.h"
#include "tensorflow/lite/kernels/kernel_util.h"
#include "tensorflow/lite/util.h"
namespace tflite {
namespace ops {
namespace builtin {
namespace stablehlo_composite {
struct State { … };
void* Init(TfLiteContext* context, const char* options, size_t options_len) { … }
void Free(TfLiteContext* context, void* node_data) { … }
TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus Eval_dynamic(TfLiteContext* context, TfLiteNode* node,
Subgraph* this_subgraph,
Subgraph* decomposition_subgraph) { … }
TfLiteStatus Eval_static(TfLiteContext* context, TfLiteNode* node,
Subgraph* this_subgraph,
Subgraph* decomposition_subgraph) { … }
TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_STABLEHLO_COMPOSITE() { … }
}
}
}