#include <BatchToSpaceND.h>

Collaboration diagram for luci_interpreter::kernels::BatchToSpaceND:

Public Member Functions
	BatchToSpaceND (const Tensor input, const Tensor block_shape, const Tensor crops, Tensor output)

const Tensor *	input () const

const Tensor *	block_shape () const

const Tensor *	crops () 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 BatchToSpaceND.h.

Constructor & Destructor Documentation

◆ BatchToSpaceND()

luci_interpreter::kernels::BatchToSpaceND::BatchToSpaceND	(	const Tensor *	input,
		const Tensor *	block_shape,
		const Tensor *	crops,
		Tensor *	output
	)

Definition at line 37 of file BatchToSpaceND.cpp.

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

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

Member Function Documentation

◆ block_shape()

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

inline

Definition at line 34 of file BatchToSpaceND.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 BatchToSpaceND(), configure(), and execute().

◆ configure()

void luci_interpreter::kernels::BatchToSpaceND::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 43 of file BatchToSpaceND.cpp.

{
 
  const auto *block_shape_data = block_shape()->data<int32_t>();
  const auto *crops_data = crops()->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(crops()->shape().num_dims() == 2);
  LUCI_INTERPRETER_CHECK(crops()->shape().dim(0) == spatial_dims_num);
  LUCI_INTERPRETER_CHECK(crops()->shape().dim(1) == 2);
  for (int i = 0; i < spatial_dims_num * 2; ++i)
  {
    LUCI_INTERPRETER_CHECK(crops_data[i] >= 0);
  }
 
  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)
  {
    LUCI_INTERPRETER_CHECK(output_batch_size % block_shape_data[i] == 0);
    output_batch_size = output_batch_size / block_shape_data[i];
    output_shape.dim(i + 1) =
      input()->shape().dim(i + 1) * block_shape_data[i] - crops_data[i * 2] - crops_data[i * 2 + 1];
  }
 
  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(), crops(), luci_interpreter::Tensor::data(), luci_interpreter::Shape::dim(), input(), LUCI_INTERPRETER_CHECK, luci::must_cast(), luci_interpreter::Shape::num_dims(), output(), output_shape, luci_interpreter::Tensor::resize(), and luci_interpreter::Tensor::shape().

◆ crops()

const Tensor * luci_interpreter::kernels::BatchToSpaceND::crops ( ) const

inline

Definition at line 35 of file BatchToSpaceND.h.

35{ return _inputs[2]; }

References luci_interpreter::Kernel::_inputs.

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

◆ execute()

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

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 81 of file BatchToSpaceND.cpp.

{
  switch (input()->element_type())
  {
    case DataType::FLOAT32:
      luci_interpreter_pal::BatchToSpaceND(
        getTensorShape(input()), getTensorData<float>(input()), getTensorShape(block_shape()),
        getTensorData<int32_t>(block_shape()), getTensorShape(crops()),
        getTensorData<int32_t>(crops()), getTensorShape(output()), getTensorData<float>(output()));
      break;
    case DataType::U8:
      luci_interpreter_pal::BatchToSpaceND(
        getTensorShape(input()), getTensorData<uint8_t>(input()), getTensorShape(block_shape()),
        getTensorData<int32_t>(block_shape()), getTensorShape(crops()),
        getTensorData<int32_t>(crops()), getTensorShape(output()),
        getTensorData<uint8_t>(output()));
      break;
    default:
      throw std::runtime_error("luci-intp BatchToSpaceND Unsupported type.");
  }
}