19#include "kernels/Utils.h"
22#include "PALSpaceToBatchND.h"
33 Tensor::element_type(kernel.
output()));
46 const int32_t pad_value = 0;
47 switch (Tensor::element_type(kernel.
input1()))
50 case DataType::FLOAT32:
52 luci_interpreter_pal::SpaceToBatchND(
65 assert(
false &&
"Unsupported type");
uint8_t * getConstDataByTensor(const circle::Tensor *raw_tensor)
uint8_t * getDataByTensor(const circle::Tensor *raw_tensor)
const circle::Tensor * input1() const
const circle::Tensor * input2() const
const circle::Tensor * input3() const
const circle::Tensor * output() const
#define LUCI_INTERPRETER_CHECK(cond)
luci_interpreter::RuntimeShape getTensorRuntimeShape(const circle::Tensor *circle_tensor, BaseRuntimeGraph *runtime_graph)
void configure_kernel_CircleSpaceToBatchND(const circle::Operator *cur_op, BaseRuntimeGraph *runtime_graph)
void execute_kernel_CircleSpaceToBatchND(const circle::Operator *cur_op, BaseRuntimeGraph *runtime_graph)