#include <SpaceToBatchND.h>

Collaboration diagram for luci_interpreter::kernels::SpaceToBatchND:

Public Member Functions
	SpaceToBatchND (const Tensor input, const Tensor block_shape, const Tensor paddings, Tensor output)

const Tensor *	input () const

const Tensor *	block_shape () const

const Tensor *	paddings () const

Tensor *	output () const

void	configure () override

void	execute () const override

Public Member Functions inherited from luci_interpreter::Kernel
virtual	~Kernel ()=default

const std::vector< const Tensor * > &	getInputTensors () const

const std::vector< Tensor * > &	getOutputTensors () const

Additional Inherited Members
Protected Member Functions inherited from luci_interpreter::Kernel
	Kernel (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs)

Protected Attributes inherited from luci_interpreter::Kernel
const std::vector< const Tensor * >	_inputs

const std::vector< Tensor * >	_outputs

Detailed Description

Definition at line 27 of file SpaceToBatchND.h.

Constructor & Destructor Documentation

◆ SpaceToBatchND()

luci_interpreter::kernels::SpaceToBatchND::SpaceToBatchND	(	const Tensor *	input,
		const Tensor *	block_shape,
		const Tensor *	paddings,
		Tensor *	output
	)

Definition at line 37 of file SpaceToBatchND.cpp.

  : Kernel({input, block_shape, paddings}, {output})
{
}

References block_shape(), input(), and paddings().

Member Function Documentation

◆ block_shape()

const Tensor * luci_interpreter::kernels::SpaceToBatchND::block_shape ( ) const

inline

Definition at line 34 of file SpaceToBatchND.h.

34{ return _inputs[1]; }

luci_interpreter::Kernel::_inputs

const std::vector< const Tensor * > _inputs

Definition Kernel.h:52

References luci_interpreter::Kernel::_inputs.

Referenced by configure(), execute(), and SpaceToBatchND().

◆ configure()

void luci_interpreter::kernels::SpaceToBatchND::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 43 of file SpaceToBatchND.cpp.

{
  const auto *block_shape_data = block_shape()->data<int32_t>();
  const auto *paddings_data = paddings()->data<int32_t>();
  LUCI_INTERPRETER_CHECK(input()->shape().num_dims() >= kInputMinDimensionNum);
  LUCI_INTERPRETER_CHECK(input()->shape().num_dims() <= kInputMaxDimensionNum);
  LUCI_INTERPRETER_CHECK(input()->element_type() == output()->element_type());
 
  int spatial_dims_num = input()->shape().num_dims() - 2;
 
  LUCI_INTERPRETER_CHECK(block_shape()->shape().num_dims() == 1);
  LUCI_INTERPRETER_CHECK(block_shape()->shape().dim(0) == spatial_dims_num);
 
  LUCI_INTERPRETER_CHECK(paddings()->shape().num_dims() == 2);
  LUCI_INTERPRETER_CHECK(paddings()->shape().dim(0) == spatial_dims_num);
  LUCI_INTERPRETER_CHECK(paddings()->shape().dim(1) == 2);
 
  Shape output_shape = Shape(input()->shape().num_dims());
  int output_batch_size = input()->shape().dim(0);
  for (int i = 0; i < spatial_dims_num; ++i)
  {
    int final_dim_size =
      (input()->shape().dim(i + 1) + paddings_data[i * 2] + paddings_data[i * 2 + 1]);
    LUCI_INTERPRETER_CHECK(final_dim_size % block_shape_data[i] == 0);
    output_shape.dim(i + 1) = final_dim_size / block_shape_data[i];
    output_batch_size = output_batch_size * block_shape_data[i];
  }
  output_shape.dim(0) = output_batch_size;
  output_shape.dim(input()->shape().num_dims() - 1) =
    input()->shape().dim(input()->shape().num_dims() - 1);
  output()->resize(output_shape);
}

References block_shape(), luci_interpreter::Tensor::data(), luci_interpreter::Shape::dim(), input(), LUCI_INTERPRETER_CHECK, luci::must_cast(), luci_interpreter::Shape::num_dims(), output(), output_shape, paddings(), luci_interpreter::Tensor::resize(), and luci_interpreter::Tensor::shape().

◆ execute()

void luci_interpreter::kernels::SpaceToBatchND::execute ( ) const

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 76 of file SpaceToBatchND.cpp.

{
  switch (input()->element_type())
  {
    tflite::SpaceToBatchParams op_params;
    case DataType::FLOAT32:
      op_params.output_offset = 0;
      luci_interpreter_pal::SpaceToBatchND(
        op_params, getTensorShape(input()), getTensorData<float>(input()),
        getTensorShape(block_shape()), getTensorData<int32_t>(block_shape()),
        getTensorShape(paddings()), getTensorData<int32_t>(paddings()), getTensorShape(output()),
        getTensorData<float>(output()));
      break;
    case DataType::U8:
      op_params.output_offset = output()->zero_point();
      luci_interpreter_pal::SpaceToBatchND(
        op_params, getTensorShape(input()), getTensorData<uint8_t>(input()),
        getTensorShape(block_shape()), getTensorData<int32_t>(block_shape()),
        getTensorShape(paddings()), getTensorData<int32_t>(paddings()), getTensorShape(output()),
        getTensorData<uint8_t>(output()));
      break;
    default:
      throw std::runtime_error("luci-intp ShapeToBatchND Unsupported type.");
  }
}