#include <array>
#include <cstddef>
#include <cstdint>
#include "tensorflow/lite/core/c/builtin_op_data.h"
#include "tensorflow/lite/core/c/common.h"
#include "tensorflow/lite/kernels/internal/tensor_ctypes.h"
#include "tensorflow/lite/kernels/kernel_util.h"
#include "tensorflow/lite/kernels/rng_util.h"
namespace tflite {
namespace ops {
namespace builtin {
namespace stablehlo_rng_bit_generator {
namespace {
constexpr int kInitialState = …;
constexpr int kOutputKey = …;
constexpr int kOutput = …;
template <typename T, size_t K>
void FillOutputBuffer(uint32_t* output_buffer, uint32_t* output_state_buffer,
int64_t output_num_elements, T fn,
std::array<uint32_t, K>& ctr, uint32_t key_0,
uint32_t key_1) { … }
}
TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_STABLEHLO_RNG_BIT_GENERATOR() { … }
}
}
}