55 thrsh.
configure(&src, &dst, threshold, false_value, true_value, type, upper);
59 validate(dst.info()->valid_region(), valid_region);
Basic function to run CLThresholdKernel.
Data set containing large tensor shapes.
ValidRegion shape_to_valid_region(TensorShape shape, bool border_undefined=false, BorderSize border_size=BorderSize(0))
Create a valid region based on tensor shape, border mode and border size.
bool is_resizable() const override
Flag indicating whether the size of the tensor can be changed.
Container for 2D border size.
Data set containing small tensor shapes.
FIXTURE_DATA_TEST_CASE(RunSmall, CLActivationLayerFixture< half >, framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallShapes(), ActivationDataset), framework::dataset::make("DataType", DataType::F16)))
combine(concat(datasets::Small2DShapes(), datasets::Large2DShapes()), framework::dataset::make("DataType", DataType::S16))
std::enable_if< is_container< T >::value, ContainerDataset< T > >::type make(std::string name, T &&values)
Helper function to create a ContainerDataset.
PaddingSize padding() const override
Padding of tensor.
TensorInfo * info() const override
Interface to be implemented by the child class to return the tensor's metadata.
DatasetMode
Possible dataset modes.
ARM_COMPUTE_EXPECT(shape.num_dimensions()==dimension+1, framework::LogLevel::ERRORS)
Accessor implementation for CLTensor objects.
TEST_SUITE_END() DATA_TEST_CASE(Configuration
void validate(const arm_compute::ValidRegion ®ion, const arm_compute::ValidRegion &reference)
Validate valid regions.
#define TEST_SUITE(SUITE_NAME)
JoinDataset< T, U > concat(T &&dataset1, U &&dataset2)
Helper function to create a JoinDataset.
DataType
Available data types.
template SimpleTensor< uint8_t > threshold(const SimpleTensor< uint8_t > &src, uint8_t threshold, uint8_t false_value, uint8_t true_value, ThresholdType type, uint8_t upper)
DATA_TEST_CASE(Configuration, framework::DatasetMode::ALL, combine(combine(concat(datasets::SmallShapes(), datasets::LargeShapes()), CNNDataTypes), framework::dataset::make("InPlace",{false, true})), shape, data_type, in_place)
Calculate required padding.
void configure(const ICLTensor *input, ICLTensor *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.
Basic implementation of the OpenCL tensor interface.