#include <NEGatherEx.h>

Collaboration diagram for arm_compute::NEGatherEx:

Public Member Functions
void	configure (const ITensor input, const ITensor indices, ITensor *output, int axis=0)

Static Public Member Functions
static Status	validate (const ITensorInfo input, const ITensorInfo indices, const ITensorInfo *output, int axis)

Detailed Description

Basic function to run NEGatherKernelEx

Definition at line 53 of file NEGatherEx.h.

Member Function Documentation

void arm_compute::NEGatherEx::configure	(	const ITensor *	input,
		const ITensor *	indices,
		ITensor *	output,
		int	axis = `0`
	)

Initialise the kernel's inputs and outputs

Parameters

[in]	input	Source tensor. Supported tensor rank: up to 4. Data type supported: U8/S8/QASYMM8/U16/S16/U32/S32/F16/F32
[in]	indices	Indices tensor. Supported tensor rank: up to 3. Must be one of the following type: U32/S32. Each value Must be in range [0, input.shape[`axis`])
[out]	output	Destination tensor. Data type supported: Same as `input`
[in]	axis	(Optional) The axis in `input` to gather `indices` from. Defaults to 0

Definition at line 49 of file NEGatherEx.cpp.

{
  auto k = std::make_unique<NEGatherKernelEx>();
  k->configure(input, indices, output, axis);
  _kernel = std::move(k);
}

Status arm_compute::NEGatherEx::validate	(	const ITensorInfo *	input,
		const ITensorInfo *	indices,
		const ITensorInfo *	output,
		int	axis
	)

static

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

Parameters

[in]	input	Source tensor info. Supported tensor rank: up to 4. Data type supported: U8/S8/QASYMM8/U16/S16/U32/S32/F16/F32
[in]	indices	Indices tensor info. Supported tensor rank: up to 3. Must be one of the following types: U32/S32. Each value Must be in range [0, input.shape[`axis`])
[in]	output	Destination tensor info. Data type supported: Same as `input`
[in]	axis	(Optional) The axis in `input` to gather `indices` from. Defaults to 0

Definition at line 56 of file NEGatherEx.cpp.

{
  return NEGatherKernelEx::validate(input, indices, output, axis);
}

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