ARM Compute Library
17.04
|
Basic function to run NEThresholdKernel. More...
#include <NEThreshold.h>
Public Member Functions | |
void | configure (const ITensor *input, ITensor *output, uint8_t threshold, uint8_t false_value=0, uint8_t true_value=0, ThresholdType type=ThresholdType::BINARY, uint8_t upper=0) |
Initialise the function's source, destination, thresholds and threshold type. More... | |
Public Member Functions inherited from INESimpleFunction | |
INESimpleFunction () | |
Constructor. More... | |
void | run () overridefinal |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
Basic function to run NEThresholdKernel.
Definition at line 37 of file NEThreshold.h.
void configure | ( | const ITensor * | input, |
ITensor * | output, | ||
uint8_t | threshold, | ||
uint8_t | false_value = 0 , |
||
uint8_t | true_value = 0 , |
||
ThresholdType | type = ThresholdType::BINARY , |
||
uint8_t | upper = 0 |
||
) |
Initialise the function's source, destination, thresholds and threshold type.
[in] | input | First tensor input. Data type supported: U8. |
[out] | output | Output tensor. Data type supported: U8. |
[in] | threshold | Threshold. If upper threshold is specified, this will be used as the lower threshold |
[in] | false_value | Value to assign when the condition is false |
[in] | true_value | value to assign when the condition is true |
[in] | type | Thresholding type. Can either be BINARY or RANGE. |
[in] | upper | Upper threshold. Only used with RANGE thresholding |