ONE - On-device Neural Engine
Loading...
Searching...
No Matches
arm_compute::NECastBool Class Reference

Class to run INESimpleFunctionNoBorder. More...

#include <NECastBool.h>

Collaboration diagram for arm_compute::NECastBool:

Public Member Functions

void configure (const ITensor *input, ITensor *output)
 

Static Public Member Functions

static Status validate (const ITensorInfo *input, const ITensorInfo *output)
 

Detailed Description

Class to run INESimpleFunctionNoBorder.

Definition at line 54 of file NECastBool.h.

Member Function Documentation

◆ configure()

void NECastBool::configure ( const ITensor *  input,
ITensor *  output 
)

Initialize the function's source, destination

Valid conversions Input -> Output :

  • U8 -> U8, S8, U16, S16, U32, S32, F32, F16
Parameters
[in]inputThe input tensor to convert. Data types supported: U8
[out]outputThe output tensor. Data types supported: U8/S8/U16/S16/U32/S32/F16/F32.

Definition at line 46 of file NECastBool.cpp.

47{
48 auto k = std::make_unique<NECastBoolKernel>();
49 k->configure(input, output);
50 _kernel = std::move(k);
51}

◆ validate()

Status NECastBool::validate ( const ITensorInfo *  input,
const ITensorInfo *  output 
)
static

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

Parameters
[in]inputSource tensor info. Data types supported: U8.
[in]outputDestination tensor info. Data type supported: U8/S8/U16/S16/U32/S32/F16/F32.
Returns
a status

Definition at line 53 of file NECastBool.cpp.

54{
55 return NECastBoolKernel::validate(input, output);
56}
static Status validate(const ITensorInfo *input, const ITensorInfo *output)

References arm_compute::NECastBoolKernel::validate().


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