#include <Squeeze.h>

Collaboration diagram for luci_interpreter::kernels::Squeeze:

[legend]

Public Member Functions
	Squeeze (const Tensor input, Tensor output, const SqueezeParams &params)

const Tensor *	input () const

Tensor *	output () const

void	configure () override

void	execute () const override

	Squeeze (const Tensor input, Tensor output, const SqueezeParams &params)

const Tensor *	input () const

Tensor *	output () const

void	configure () override

void	execute () const override

Public Member Functions inherited from luci_interpreter::KernelWithParams< SqueezeParams >
const SqueezeParams &	params () const

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::KernelWithParams< SqueezeParams >
	KernelWithParams (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs, const SqueezeParams &params)

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

Protected Attributes inherited from luci_interpreter::KernelWithParams< SqueezeParams >
const SqueezeParams	_params

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

const std::vector< Tensor * >	_outputs

Detailed Description

Definition at line 29 of file Squeeze.h.

Constructor & Destructor Documentation

◆ Squeeze() [1/2]

luci_interpreter::kernels::Squeeze::Squeeze	(	const Tensor *	input,
		Tensor *	output,
		const SqueezeParams &	params
	)

Definition at line 29 of file Squeeze.cpp.

  : KernelWithParams<SqueezeParams>({input}, {output}, params)
{
}

References input().

◆ Squeeze() [2/2]

luci_interpreter::kernels::Squeeze::Squeeze	(	const Tensor *	input,
		Tensor *	output,
		const SqueezeParams &	params
	)

Member Function Documentation

◆ configure() [1/2]

void luci_interpreter::kernels::Squeeze::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 34 of file Squeeze.cpp.

{
  int input_num_dims = input()->shape().num_dims();
  int num_squeeze_dims = params().squeeze_dims.size();
  assert(input_num_dims <= 8);
  bool should_squeeze[8] = {false};
  int num_squeezed_dims = 0;
  if (num_squeeze_dims == 0)
  {
    for (int idx = 0; idx < input_num_dims; ++idx)
    {
      if (input()->shape().dim(idx) == 1)
      {
        should_squeeze[idx] = true;
        ++num_squeezed_dims;
      }
    }
  }
  else
  {
    for (int idx = 0; idx < num_squeeze_dims; ++idx)
    {
      int current = params().squeeze_dims[idx] < 0 ? params().squeeze_dims[idx] + input_num_dims
                                                   : params().squeeze_dims[idx];
      assert(current >= 0 && current < input_num_dims && input()->shape().dim(current) == 1);
      if (!should_squeeze[current])
        ++num_squeezed_dims;
      should_squeeze[current] = true;
    }
  }
  Shape output_shape(input_num_dims - num_squeezed_dims);
  for (int in_idx = 0, out_idx = 0; in_idx < input_num_dims; ++in_idx)
  {
    if (!should_squeeze[in_idx])
    {
      output_shape.dim(out_idx++) = input()->shape().dim(in_idx);
    }
  }
  output()->resize(output_shape);
}

References luci_interpreter::Shape::dim(), input(), luci::must_cast(), luci_interpreter::Shape::num_dims(), output(), output_shape, luci_interpreter::KernelWithParams< SqueezeParams >::params(), luci_interpreter::Tensor::resize(), luci_interpreter::Tensor::shape(), and luci_interpreter::SqueezeParams::squeeze_dims.

◆ configure() [2/2]

void luci_interpreter::kernels::Squeeze::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

◆ execute() [1/2]

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

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 75 of file Squeeze.cpp.

{
  assert(input()->shape().num_elements() == output()->shape().num_elements());
 
  const auto *input_data = input()->data<void>();
  auto *output_data = output()->data<void>();
  std::memcpy(output_data, input_data,
              getDataTypeSize(input()->element_type()) * input()->shape().num_elements());
}