17#include "kernels/Pow.h"
18#include "kernels/Utils.h"
20#include <tensorflow/lite/kernels/internal/reference/reference_ops.h>
44 switch (
input1()->element_type())
46 case DataType::FLOAT32:
53 throw std::runtime_error(
"luci-intp Pow Unsupported type.");
57template <
typename T>
void Pow::eval()
const
59 tflite::ArithmeticParams params{};
61 const bool need_broadcast = tflite::reference_ops::ProcessBroadcastShapes(
void resize(const Shape &new_shape)
void execute() const override
const Tensor * input2() const
void configure() override
Pow(const Tensor *input1, const Tensor *input2, Tensor *output)
const Tensor * input1() const
#define LUCI_INTERPRETER_CHECK(cond)
Shape calculateShapeForBroadcast(const Shape &input1_shape, const Shape &input2_shape)
tflite::RuntimeShape getTensorShape(const Tensor *tensor)
void BroadcastPow4DSlow(const RuntimeShape &unextended_input1_shape, const T *input1_data, const RuntimeShape &unextended_input2_shape, const T *input2_data, const RuntimeShape &unextended_output_shape, T *output_data)
void Pow(const RuntimeShape &input1_shape, const T *input1_data, const RuntimeShape &input2_shape, const T *input2_data, const RuntimeShape &output_shape, T *output_data)