17#include "kernels/LocalResponseNormalization.h"
19#include "kernels/Utils.h"
30 const Tensor *input,
Tensor *output,
const LocalResponseNormalizationParams ¶ms)
31 : KernelWithParams<LocalResponseNormalizationParams>({
input}, {
output}, params)
35void LocalResponseNormalization::configure()
44void LocalResponseNormalization::execute()
const
46 switch (
output()->element_type())
48 case DataType::FLOAT32:
49 tflite::LocalResponseNormalizationParams op_params;
50 op_params.range = params().radius;
51 op_params.bias = params().bias;
52 op_params.alpha = params().alpha;
53 op_params.beta = params().beta;
54 luci_interpreter_pal::LocalResponseNormalization(
56 getTensorData<float>(
output()));
59 assert(
false &&
"Unsupported type.");
LocalResponseNormalization(const Tensor *input, Tensor *output, const LocalResponseNormalizationParams ¶ms)
#define LUCI_INTERPRETER_CHECK(cond)
tflite::RuntimeShape getTensorShape(const Tensor *tensor)