arm_compute::NEOneHotKernel Class Reference

#include <NEOneHotKernel.h>

Collaboration diagram for arm_compute::NEOneHotKernel:

Public Member Functions
	NEOneHotKernel ()
	NEOneHotKernel (const NEOneHotKernel &)=delete
NEOneHotKernel &	operator= (const NEOneHotKernel &)=delete
	NEOneHotKernel (NEOneHotKernel &&)=default
NEOneHotKernel &	operator= (NEOneHotKernel &&)=default
	~NEOneHotKernel ()=default
const char *	name () const override
void	configure (const ITensor *indices, const ITensor *depth, const ITensor *on_value, const ITensor *off_value, ITensor *output, int axis=-1)
void	run (const Window &window, const ThreadInfo &info) override

Static Public Member Functions
static Status	validate (const ITensorInfo *indices, const ITensorInfo *depth, const ITensorInfo *on_value, const ITensorInfo *off_value, const ITensorInfo *output, int axis=-1)

Detailed Description

Kernel to perform other operation on NEON

Definition at line 49 of file NEOneHotKernel.h.

Constructor & Destructor Documentation

NEOneHotKernel() [1/3]

arm_compute::NEOneHotKernel::NEOneHotKernel

(

)

Default constructor.

Definition at line 107 of file NEOneHotKernel.cpp.

  : _indices{nullptr}, _depth{nullptr}, _on_value{nullptr}, _off_value{nullptr}, _axis{-1},
    _output{nullptr}, _func{}
{
}

NEOneHotKernel() [2/3]

arm_compute::NEOneHotKernel::NEOneHotKernel ( const NEOneHotKernel &  )

delete

Prevent instances of this class from being copied (As this class contains pointers).

NEOneHotKernel() [3/3]

arm_compute::NEOneHotKernel::NEOneHotKernel ( NEOneHotKernel &&  )

default

Allow instances of this class to be moved.

~NEOneHotKernel()

arm_compute::NEOneHotKernel::~NEOneHotKernel ( )

default

Default detructor

Member Function Documentation

configure()

void arm_compute::NEOneHotKernel::configure	(	const ITensor *	indices,
		const ITensor *	depth,
		const ITensor *	on_value,
		const ITensor *	off_value,
		ITensor *	output,
		int	axis = -1
	)

Initialise the kernel's inputs and outputs

Parameters

[in]	indices	Indices tensor. Supported tensor rank: up to 3. Must be one of the following types: U32/S32
[in]	depth	The tensor for depth of the one hot dimension. Supported tensor rank: up to 3. Must be one of the following types: U32/S32
[in]	on_value	On value tensor. Supported tensor rank: only 1. Data type supported: U8/S8/U16/S16/F16/U32/S32/F32
[in]	off_value	Off value tensor. Supported tensor rank: only 1. Data type supported: Same as on_value
[out]	output	Destination tensor. Data type supported: Same as on_value
[in]	axis	(Optional) The axis to fill. Negative values wrap around. Defaults to -1. The value must be in range [-indices.rank , indices.rank)

Definition at line 167 of file NEOneHotKernel.cpp.

{
  ARM_COMPUTE_ERROR_ON_NULLPTR(indices, depth, on_value, off_value, output);
  ARM_COMPUTE_ERROR_ON(output->info()->total_size() == 0);
  ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(indices->info(), depth->info(), on_value->info(),
                                                off_value->info(), output->info(), axis));
  _indices = indices;
  _depth = depth;
  _on_value = on_value;
  _off_value = off_value;
  _output = output;
  _axis = wrap_around(axis, static_cast<int>(output->info()->num_dimensions()));
  if (0 == _axis)
  {
    switch (_indices->info()->data_type())
    {
      case DataType::U32:
        _func = &NEOneHotKernel::onehot_0_axis<uint32_t>;
        break;
      case DataType::S32:
        _func = &NEOneHotKernel::onehot_0_axis<int32_t>;
        break;
      default:
        ARM_COMPUTE_ERROR("Not supported");
        break;
    }
  }
  else
  {
    switch (_indices->info()->data_type())
    {
      case DataType::U32:
        _func = &NEOneHotKernel::onehot_n_axis<uint32_t>;
        break;
      case DataType::S32:
        _func = &NEOneHotKernel::onehot_n_axis<int32_t>;
        break;
      default:
        ARM_COMPUTE_ERROR("Not supported");
        break;
    }
  }
  // Create window
  Window win = calculate_max_window(*output->info(), Steps());
  output->info()->set_valid_region(ValidRegion(Coordinates(), output->info()->tensor_shape()));
  INEKernel::configure(win);
}

name()

const char * arm_compute::NEOneHotKernel::name ( ) const

inlineoverride

Name of the kernel

Returns

Kernel name

Definition at line 68 of file NEOneHotKernel.h.

{ return "NEOneHotKernel"; }

operator=() [1/2]

NEOneHotKernel & arm_compute::NEOneHotKernel::operator= ( const NEOneHotKernel &  )

delete

Prevent instances of this class from being copied (As this class contains pointers).

operator=() [2/2]

NEOneHotKernel & arm_compute::NEOneHotKernel::operator= ( NEOneHotKernel &&  )

default

Allow instances of this class to be moved.

run()

void arm_compute::NEOneHotKernel::run ( const Window &  window, const ThreadInfo &  info  )

override

Definition at line 226 of file NEOneHotKernel.cpp.

{
  ARM_COMPUTE_UNUSED(info);
  ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this);
  ARM_COMPUTE_ERROR_ON(_func == nullptr);
  (this->*_func)(window, info);
}

References info.

Referenced by package.infer.session::inference().

validate()

Status arm_compute::NEOneHotKernel::validate ( const ITensorInfo *  indices, const ITensorInfo *  depth, const ITensorInfo *  on_value, const ITensorInfo *  off_value, const ITensorInfo *  output, int  axis = -1  )

static

Static function to check if given info will lead to a valid configuration of NEOneHotKernel

Parameters

[in]	indices	Indices tensor info. Supported tensor rank: up to 3. Must be one of the following types: U32/S32
[in]	depth	The tensor info for depth of the one hot dimension. Supported tensor rank: up to 3. Must be one of the following types: U32/S32
[in]	on_value	On value tensor info. Supported tensor rank: only 1. Data type supported: U8/S8/U16/S16/F16/U32/S32/F32
[in]	off_value	Off value tensor info. Supported tensor rank: only 1. Data type supported: Same as on_value
[out]	output	Destination tensor info. Data type supported: Same as on_value
[in]	axis	(Optional) The axis to fill. Negative values wrap around. Defaults to -1. The value must be in range [-indices.rank , indices.rank)

Returns

a status

Definition at line 217 of file NEOneHotKernel.cpp.

{
  ARM_COMPUTE_RETURN_ON_ERROR(
    validate_arguments(indices, depth, on_value, off_value, output, axis));
  return Status{};
}

Referenced by arm_compute::NEOneHot::validate().

---

The documentation for this class was generated from the following files:

• compute/ARMComputeEx/arm_compute/core/NEON/kernels/NEOneHotKernel.h
• compute/ARMComputeEx/src/core/NEON/kernels/NEOneHotKernel.cpp