ONE - On-device Neural Engine
•All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Public Member Functions | Static Public Member Functions
arm_compute::CLGatherEx Class Reference

Class to to run CLGatherKernel. More...

#include <CLGatherEx.h>

Collaboration diagram for arm_compute::CLGatherEx:
Collaboration graph
[legend]

Public Member Functions

void configure (const ICLTensor *input, const ICLTensor *indices, ICLTensor *output, int axis=0)
 Initialise the kernel's inputs, output and convertion policy.
 

Static Public Member Functions

static Status validate (const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, int axis=0)
 Static function to check if given info will lead to a valid configuration of CLGatherEx.
 

Detailed Description

Class to to run CLGatherKernel.

Definition at line 62 of file CLGatherEx.h.

Member Function Documentation

◆ configure()

void CLGatherEx::configure ( const ICLTensor *  input,
const ICLTensor *  indices,
ICLTensor *  output,
int  axis = 0 
)

Initialise the kernel's inputs, output and convertion policy.

Parameters
[in]inputAn input tensor. Data types supported: U8/QASYMM8/S32/F32.
[in]indicesAn indexes tensor. Data types supported: S32.
[out]outputThe output tensor, Data types supported: same as input.
[in]axis(Optional) The axis in input to gather indices from. Defaults to 0
Returns
N/A

Definition at line 50 of file CLGatherEx.cpp.

52{
53 auto k = std::make_unique<CLGatherExKernel>();
54 k->configure(input, indices, output, axis);
55 _kernel = std::move(k);
56}

◆ validate()

Status CLGatherEx::validate ( const ITensorInfo *  input,
const ITensorInfo *  indices,
const ITensorInfo *  output,
int  axis = 0 
)
static

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

Parameters
[in]inputAn input tensor. Data types supported: U8/QASYMM8/S32/F32.
[in]indicesAn indexes tensor. Data types supported: S32.
[out]outputThe output tensor, Data types supported: same as input.
[in]axis(Optional) The axis in input to gather indices from. Defaults to 0
Returns
a status

Definition at line 58 of file CLGatherEx.cpp.

60{
61 return CLGatherExKernel::validate(input, indices, output, axis);
62}
arm_compute::CLGatherExKernel::validate
static Status validate(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, int axis=0)
Static function to check if given info will lead to a valid configuration of CLGatherExKernel.
Definition CLGatherExKernel.cpp:144

References arm_compute::CLGatherExKernel::validate().

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