#include <algorithm>
#include <cmath>
#include <cstdint>
#include <limits>
#include <random>
#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 custom {
namespace random_uniform {
struct OpData { … };
namespace {
template <typename T, typename dist_type>
void RandomUniformSample(std::default_random_engine& rng, T* buffer,
size_t buffer_size, T min_value, T max_value) { … }
TfLiteIntArray* CreateDimensionsFromTensor(const TfLiteTensor* tensor) { … }
}
void* Init(TfLiteContext* context, const char* buffer, size_t length) { … }
void Free(TfLiteContext* context, void* buffer) { … }
TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { … }
TfLiteStatus EvalFloat(TfLiteContext* context, TfLiteNode* node) { … }
int64_t IntValueFromTensor(const TfLiteTensor* tensor) { … }
TfLiteStatus EvalInt(TfLiteContext* context, TfLiteNode* node) { … }
}
TfLiteRegistration* Register_RANDOM_UNIFORM() { … }
TfLiteRegistration* Register_RANDOM_UNIFORM_INT() { … }
}
}
}