18#include "kernels/Utils.h"
33 output()->element_type() == DataType::S64);
53 throw std::runtime_error(
"luci-intp Shape Unsupported type.");
57template <
typename T>
void ShapeKernel::evalInt()
const
61 auto output_data = getTensorData<T>(
output());
63 for (
int i = 0; i < input_shape.num_dims(); ++i)
65 output_data[i] = input_shape.dim(i);
const ShapeParams & params() const
void resize(const Shape &new_shape)
const Shape & shape() const
ShapeKernel(const Tensor *input, Tensor *output, const ShapeParams ¶ms)
void execute() const override
void configure() override
const Tensor * input() const
#define LUCI_INTERPRETER_CHECK(cond)
const luci_interpreter::RuntimeShape output_shape