21.02
|
Enumerations | |
enum | ArithmeticOperation { ADD, SUB } |
Arithmetic operation types. More... | |
enum | FFTDirection { Forward, Inverse } |
enum | WinogradTransformType { INPUT, FILTER, OUTPUT } |
Winograd transform type. More... | |
Functions | |
template<typename T > | |
SimpleTensor< T > | absolute_difference (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2, DataType dst_data_type) |
template SimpleTensor< uint8_t > | absolute_difference (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2, DataType dst_data_type) |
template SimpleTensor< int16_t > | absolute_difference (const SimpleTensor< int16_t > &src1, const SimpleTensor< int16_t > &src2, DataType dst_data_type) |
template SimpleTensor< int8_t > | absolute_difference (const SimpleTensor< int8_t > &src1, const SimpleTensor< int8_t > &src2, DataType dst_data_type) |
template<typename T1 , typename T2 > | |
SimpleTensor< T2 > | accumulate (const SimpleTensor< T1 > &src, DataType output_data_type) |
template<typename T1 , typename T2 > | |
SimpleTensor< T2 > | accumulate_weighted (const SimpleTensor< T1 > &src, float alpha, DataType output_data_type) |
template<typename T1 , typename T2 > | |
SimpleTensor< T2 > | accumulate_squared (const SimpleTensor< T1 > &src, uint32_t shift, DataType output_data_type) |
template SimpleTensor< int16_t > | accumulate (const SimpleTensor< uint8_t > &src, DataType output_data_type) |
template SimpleTensor< uint8_t > | accumulate_weighted (const SimpleTensor< uint8_t > &src, float alpha, DataType output_data_type) |
template SimpleTensor< int16_t > | accumulate_squared (const SimpleTensor< uint8_t > &src, uint32_t shift, DataType output_data_type) |
template<typename T > | |
SimpleTensor< T > | activation_layer (const SimpleTensor< T > &src, ActivationLayerInfo info, const QuantizationInfo &oq_info) |
template<> | |
SimpleTensor< uint8_t > | activation_layer< uint8_t > (const SimpleTensor< uint8_t > &src, ActivationLayerInfo info, const QuantizationInfo &oq_info) |
template<> | |
SimpleTensor< int8_t > | activation_layer< int8_t > (const SimpleTensor< int8_t > &src, ActivationLayerInfo info, const QuantizationInfo &oq_info) |
template<> | |
SimpleTensor< int16_t > | activation_layer< int16_t > (const SimpleTensor< int16_t > &src, ActivationLayerInfo info, const QuantizationInfo &oq_info) |
template SimpleTensor< int32_t > | activation_layer (const SimpleTensor< int32_t > &src, ActivationLayerInfo info, const QuantizationInfo &oq_info) |
template SimpleTensor< float > | activation_layer (const SimpleTensor< float > &src, ActivationLayerInfo info, const QuantizationInfo &oq_info) |
template SimpleTensor< half > | activation_layer (const SimpleTensor< half > &src, ActivationLayerInfo info, const QuantizationInfo &oq_info) |
template<typename T > | |
T | activate_float (T x, T a, T b, ActivationLayerInfo::ActivationFunction activation) |
template<typename T > | |
SimpleTensor< T > | arithmetic_division (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2, DataType data_type) |
template SimpleTensor< half > | arithmetic_division (const SimpleTensor< half > &src1, const SimpleTensor< half > &src2, DataType data_type) |
template SimpleTensor< float > | arithmetic_division (const SimpleTensor< float > &src1, const SimpleTensor< float > &src2, DataType data_type) |
template<typename T > | |
SimpleTensor< T > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< T > &src1, const SimpleTensor< T > &src2, SimpleTensor< T > &dst, ConvertPolicy convert_policy) |
template<> | |
SimpleTensor< uint8_t > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2, SimpleTensor< uint8_t > &dst, ConvertPolicy convert_policy) |
template<> | |
SimpleTensor< int8_t > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< int8_t > &src1, const SimpleTensor< int8_t > &src2, SimpleTensor< int8_t > &dst, ConvertPolicy convert_policy) |
template<> | |
SimpleTensor< int16_t > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< int16_t > &src1, const SimpleTensor< int16_t > &src2, SimpleTensor< int16_t > &dst, ConvertPolicy convert_policy) |
template SimpleTensor< int32_t > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< int32_t > &src1, const SimpleTensor< int32_t > &src2, SimpleTensor< int32_t > &dst, ConvertPolicy convert_policy) |
template SimpleTensor< half > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< half > &src1, const SimpleTensor< half > &src2, SimpleTensor< half > &dst, ConvertPolicy convert_policy) |
template SimpleTensor< float > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< float > &src1, const SimpleTensor< float > &src2, SimpleTensor< float > &dst, ConvertPolicy convert_policy) |
template<typename T > | |
SimpleTensor< T > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< T > &src1, const SimpleTensor< T > &src2, DataType dst_data_type, ConvertPolicy convert_policy) |
template SimpleTensor< int32_t > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< int32_t > &src1, const SimpleTensor< int32_t > &src2, DataType dst_data_type, ConvertPolicy convert_policy) |
template SimpleTensor< int16_t > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< int16_t > &src1, const SimpleTensor< int16_t > &src2, DataType dst_data_type, ConvertPolicy convert_policy) |
template SimpleTensor< int8_t > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< int8_t > &src1, const SimpleTensor< int8_t > &src2, DataType dst_data_type, ConvertPolicy convert_policy) |
template SimpleTensor< half > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< half > &src1, const SimpleTensor< half > &src2, DataType dst_data_type, ConvertPolicy convert_policy) |
template SimpleTensor< float > | arithmetic_operation (ArithmeticOperation op, const SimpleTensor< float > &src1, const SimpleTensor< float > &src2, DataType dst_data_type, ConvertPolicy convert_policy) |
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type * > | |
SimpleTensor< T > | batch_normalization_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &mean, const SimpleTensor< T > &var, const SimpleTensor< T > &beta, const SimpleTensor< T > &gamma, float epsilon, ActivationLayerInfo act_info) |
template SimpleTensor< float > | batch_normalization_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &mean, const SimpleTensor< float > &var, const SimpleTensor< float > &beta, const SimpleTensor< float > &gamma, float epsilon, ActivationLayerInfo act_info) |
template SimpleTensor< half > | batch_normalization_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &mean, const SimpleTensor< half > &var, const SimpleTensor< half > &beta, const SimpleTensor< half > &gamma, float epsilon, ActivationLayerInfo act_info) |
template<typename T > | |
SimpleTensor< T > | batch_to_space (const SimpleTensor< T > &src, const SimpleTensor< int32_t > &block_shape, const TensorShape &dst_shape) |
template SimpleTensor< float > | batch_to_space (const SimpleTensor< float > &src, const SimpleTensor< int32_t > &block_shape, const TensorShape &dst_shape) |
template SimpleTensor< half > | batch_to_space (const SimpleTensor< half > &src, const SimpleTensor< int32_t > &block_shape, const TensorShape &dst_shape) |
template<typename T > | |
SimpleTensor< T > | bitwise_and (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2) |
template SimpleTensor< uint8_t > | bitwise_and (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2) |
template<typename T > | |
SimpleTensor< T > | bitwise_not (const SimpleTensor< T > &src) |
template SimpleTensor< uint8_t > | bitwise_not (const SimpleTensor< uint8_t > &src) |
template<typename T > | |
SimpleTensor< T > | bitwise_or (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2) |
template SimpleTensor< uint8_t > | bitwise_or (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2) |
template<typename T > | |
SimpleTensor< T > | bitwise_xor (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2) |
template SimpleTensor< uint8_t > | bitwise_xor (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2) |
template<typename T , typename TDeltas > | |
SimpleTensor< T > | bounding_box_transform (const SimpleTensor< T > &boxes, const SimpleTensor< TDeltas > &deltas, const BoundingBoxTransformInfo &info) |
template SimpleTensor< float > | bounding_box_transform (const SimpleTensor< float > &boxes, const SimpleTensor< float > &deltas, const BoundingBoxTransformInfo &info) |
template SimpleTensor< half > | bounding_box_transform (const SimpleTensor< half > &boxes, const SimpleTensor< half > &deltas, const BoundingBoxTransformInfo &info) |
template<> | |
SimpleTensor< uint16_t > | bounding_box_transform (const SimpleTensor< uint16_t > &boxes, const SimpleTensor< uint8_t > &deltas, const BoundingBoxTransformInfo &info) |
template<typename T > | |
SimpleTensor< T > | box3x3 (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | box3x3 (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | canny_edge_detector (const SimpleTensor< T > &src, int32_t upper_thresh, int32_t lower_thresh, int gradient_size, MagnitudeType norm_type, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | canny_edge_detector (const SimpleTensor< uint8_t > &src, int32_t upper_thresh, int32_t lower_thresh, int gradient_size, MagnitudeType norm_type, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
std::vector< SimpleTensor< T > > | channel_combine (const TensorShape &shape, const std::vector< SimpleTensor< T >> &image_planes, Format format) |
template std::vector< SimpleTensor< uint8_t > > | channel_combine (const TensorShape &shape, const std::vector< SimpleTensor< uint8_t >> &image_planes, Format format) |
template<typename T > | |
SimpleTensor< uint8_t > | channel_extract (const TensorShape &shape, const std::vector< SimpleTensor< T >> &tensor_planes, Format format, Channel channel) |
template SimpleTensor< uint8_t > | channel_extract (const TensorShape &shape, const std::vector< SimpleTensor< uint8_t >> &tensor_planes, Format format, Channel channel) |
template<typename T > | |
SimpleTensor< T > | channel_shuffle (const SimpleTensor< T > &src, int num_groups) |
template SimpleTensor< uint8_t > | channel_shuffle (const SimpleTensor< uint8_t > &src, int num_groups) |
template SimpleTensor< uint16_t > | channel_shuffle (const SimpleTensor< uint16_t > &src, int num_groups) |
template SimpleTensor< uint32_t > | channel_shuffle (const SimpleTensor< uint32_t > &src, int num_groups) |
template SimpleTensor< half > | channel_shuffle (const SimpleTensor< half > &src, int num_groups) |
template SimpleTensor< float > | channel_shuffle (const SimpleTensor< float > &src, int num_groups) |
template<typename T > | |
SimpleTensor< T > | col2im (const SimpleTensor< T > &src, const TensorShape &dst_shape, unsigned int num_groups) |
template SimpleTensor< float > | col2im (const SimpleTensor< float > &src, const TensorShape &dst_shape, unsigned int num_groups) |
template SimpleTensor< half > | col2im (const SimpleTensor< half > &src, const TensorShape &dst_shape, unsigned int num_groups) |
template SimpleTensor< uint8_t > | col2im (const SimpleTensor< uint8_t > &src, const TensorShape &dst_shape, unsigned int num_groups) |
template<typename T > | |
std::vector< SimpleTensor< T > > | color_convert (const TensorShape &shape, const std::vector< SimpleTensor< T >> &tensor_planes, Format src_format, Format dst_format) |
template std::vector< SimpleTensor< uint8_t > > | color_convert (const TensorShape &shape, const std::vector< SimpleTensor< uint8_t >> &tensor_planes, Format src_format, Format dst_format) |
template<typename T > | |
SimpleTensor< uint8_t > | compare (ComparisonOperation op, const SimpleTensor< T > &src1, const SimpleTensor< T > &src2) |
template<> | |
SimpleTensor< uint8_t > | compare (ComparisonOperation op, const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2) |
template<> | |
SimpleTensor< uint8_t > | compare (ComparisonOperation op, const SimpleTensor< int8_t > &src1, const SimpleTensor< int8_t > &src2) |
template SimpleTensor< uint8_t > | compare (ComparisonOperation op, const SimpleTensor< half > &src1, const SimpleTensor< half > &src2) |
template SimpleTensor< uint8_t > | compare (ComparisonOperation op, const SimpleTensor< float > &src1, const SimpleTensor< float > &src2) |
template<typename T > | |
SimpleTensor< T > | compute_all_anchors (const SimpleTensor< T > &anchors, const ComputeAnchorsInfo &info) |
template SimpleTensor< float > | compute_all_anchors (const SimpleTensor< float > &anchors, const ComputeAnchorsInfo &info) |
template SimpleTensor< half > | compute_all_anchors (const SimpleTensor< half > &anchors, const ComputeAnchorsInfo &info) |
template<> | |
SimpleTensor< int16_t > | compute_all_anchors (const SimpleTensor< int16_t > &anchors, const ComputeAnchorsInfo &info) |
template<typename T > | |
SimpleTensor< T > | concatenate_layer (std::vector< SimpleTensor< T >> &srcs, SimpleTensor< T > &dst, unsigned int axis) |
template SimpleTensor< float > | concatenate_layer (std::vector< SimpleTensor< float >> &srcs, SimpleTensor< float > &dst, unsigned int axis) |
template SimpleTensor< half > | concatenate_layer (std::vector< SimpleTensor< half >> &srcs, SimpleTensor< half > &dst, unsigned int axis) |
template SimpleTensor< uint8_t > | concatenate_layer (std::vector< SimpleTensor< uint8_t >> &srcs, SimpleTensor< uint8_t > &dst, unsigned int axis) |
template SimpleTensor< int8_t > | concatenate_layer (std::vector< SimpleTensor< int8_t >> &srcs, SimpleTensor< int8_t > &dst, unsigned int axis) |
template<typename T > | |
SimpleTensor< T > | convert_fully_connected_weights (const SimpleTensor< T > &src, const TensorShape &original_input_shape, const DataLayout training_data_layout) |
template SimpleTensor< uint8_t > | convert_fully_connected_weights (const SimpleTensor< uint8_t > &src, const TensorShape &original_input_shape, const DataLayout training_data_layout) |
template SimpleTensor< half > | convert_fully_connected_weights (const SimpleTensor< half > &src, const TensorShape &original_input_shape, const DataLayout training_data_layout) |
template SimpleTensor< float > | convert_fully_connected_weights (const SimpleTensor< float > &src, const TensorShape &original_input_shape, const DataLayout training_data_layout) |
template<typename T > | |
SimpleTensor< T > | convolution (const SimpleTensor< uint8_t > &src, DataType output_data_type, const int16_t *conv, uint32_t scale, BorderMode border_mode, uint8_t constant_border_value, const unsigned int width, const unsigned int height) |
template<typename T , typename TW , typename TB > | |
SimpleTensor< T > | convolution_layer_nchw (const SimpleTensor< T > &src, const SimpleTensor< TW > &weights, const SimpleTensor< TB > &bias, SimpleTensor< T > &dst, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups) |
template<typename T , typename TW , typename TB > | |
SimpleTensor< T > | convolution_layer (const SimpleTensor< T > &src, const SimpleTensor< TW > &weights, const SimpleTensor< TB > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info) |
template SimpleTensor< float > | convolution_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &weights, const SimpleTensor< float > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info) |
template SimpleTensor< half > | convolution_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &weights, const SimpleTensor< half > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info) |
template SimpleTensor< uint8_t > | convolution_layer (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info) |
template SimpleTensor< uint8_t > | convolution_layer (const SimpleTensor< uint8_t > &src, const SimpleTensor< int8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info) |
template SimpleTensor< int8_t > | convolution_layer (const SimpleTensor< int8_t > &src, const SimpleTensor< int8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info) |
template<typename T > | |
SimpleTensor< T > | copy (const SimpleTensor< T > &src, const TensorShape &output_shape) |
template SimpleTensor< uint8_t > | copy (const SimpleTensor< uint8_t > &src, const TensorShape &output_shape) |
template SimpleTensor< int8_t > | copy (const SimpleTensor< int8_t > &src, const TensorShape &output_shape) |
template SimpleTensor< uint16_t > | copy (const SimpleTensor< uint16_t > &src, const TensorShape &output_shape) |
template SimpleTensor< int16_t > | copy (const SimpleTensor< int16_t > &src, const TensorShape &output_shape) |
template SimpleTensor< uint32_t > | copy (const SimpleTensor< uint32_t > &src, const TensorShape &output_shape) |
template SimpleTensor< int32_t > | copy (const SimpleTensor< int32_t > &src, const TensorShape &output_shape) |
template SimpleTensor< half > | copy (const SimpleTensor< half > &src, const TensorShape &output_shape) |
template SimpleTensor< float > | copy (const SimpleTensor< float > &src, const TensorShape &output_shape) |
template<typename T > | |
SimpleTensor< float > | crop_and_resize (const SimpleTensor< T > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template SimpleTensor< float > | crop_and_resize (const SimpleTensor< float > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template SimpleTensor< float > | crop_and_resize (const SimpleTensor< uint16_t > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template SimpleTensor< float > | crop_and_resize (const SimpleTensor< uint32_t > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template SimpleTensor< float > | crop_and_resize (const SimpleTensor< int16_t > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template SimpleTensor< float > | crop_and_resize (const SimpleTensor< int32_t > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template SimpleTensor< float > | crop_and_resize (const SimpleTensor< half > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template SimpleTensor< float > | crop_and_resize (const SimpleTensor< uint8_t > &src, const SimpleTensor< float > &boxes, SimpleTensor< int32_t > box_ind, Coordinates2D crop_size, InterpolationPolicy method, float extrapolation_value) |
template<typename T , typename TB > | |
SimpleTensor< T > | deconvolution_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &weights, const SimpleTensor< TB > &bias, const TensorShape &output_shape, const PadStrideInfo &info, QuantizationInfo out_qinfo=QuantizationInfo()) |
Deconvolution reference implementation. More... | |
template SimpleTensor< uint8_t > | deconvolution_layer (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, QuantizationInfo out_quant_info) |
template SimpleTensor< int8_t > | deconvolution_layer (const SimpleTensor< int8_t > &src, const SimpleTensor< int8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, QuantizationInfo out_quant_info) |
template SimpleTensor< float > | deconvolution_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &weights, const SimpleTensor< float > &bias, const TensorShape &output_shape, const PadStrideInfo &info, QuantizationInfo out_quant_info) |
template SimpleTensor< half > | deconvolution_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &weights, const SimpleTensor< half > &bias, const TensorShape &output_shape, const PadStrideInfo &info, QuantizationInfo out_quant_info) |
template<typename T > | |
SimpleTensor< T > | depthconcatenate_layer (const std::vector< SimpleTensor< T >> &srcs, SimpleTensor< T > &dst) |
template SimpleTensor< uint8_t > | depthconcatenate_layer (const std::vector< SimpleTensor< uint8_t >> &srcs, SimpleTensor< uint8_t > &dst) |
template SimpleTensor< float > | depthconcatenate_layer (const std::vector< SimpleTensor< float >> &srcs, SimpleTensor< float > &dst) |
template SimpleTensor< half > | depthconcatenate_layer (const std::vector< SimpleTensor< half >> &srcs, SimpleTensor< half > &dst) |
template<typename T1 , typename T2 , typename std::enable_if< std::is_integral< T1 >::value &&!std::is_same< T1, T2 >::value, int >::type > | |
SimpleTensor< T2 > | depth_convert (const SimpleTensor< T1 > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< int8_t > | depth_convert (const SimpleTensor< uint8_t > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< uint8_t > | depth_convert (const SimpleTensor< int8_t > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< uint8_t > | depth_convert (const SimpleTensor< uint16_t > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< uint8_t > | depth_convert (const SimpleTensor< int16_t > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< uint8_t > | depth_convert (const SimpleTensor< uint32_t > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< uint8_t > | depth_convert (const SimpleTensor< int32_t > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< float > | depth_convert (const SimpleTensor< bfloat16 > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< uint8_t > | depth_convert (const SimpleTensor< half > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template SimpleTensor< uint8_t > | depth_convert (const SimpleTensor< float > &src, DataType dt_out, ConvertPolicy policy, uint32_t shift) |
template<typename T > | |
SimpleTensor< T > | depth_to_space (const SimpleTensor< T > &src, const TensorShape &dst_shape, int32_t block_shape) |
template SimpleTensor< float > | depth_to_space (const SimpleTensor< float > &src, const TensorShape &dst_shape, int32_t block_shape) |
template SimpleTensor< half > | depth_to_space (const SimpleTensor< half > &src, const TensorShape &dst_shape, int32_t block_shape) |
template<> | |
SimpleTensor< float > | depthwise_convolution (const SimpleTensor< float > &src, const SimpleTensor< float > &weights, const SimpleTensor< float > &biases, const TensorShape &dst_shape, const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const QuantizationInfo &out_quant_info) |
template<> | |
SimpleTensor< half > | depthwise_convolution (const SimpleTensor< half > &src, const SimpleTensor< half > &weights, const SimpleTensor< half > &biases, const TensorShape &dst_shape, const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const QuantizationInfo &out_quant_info) |
template<> | |
SimpleTensor< uint8_t > | depthwise_convolution (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &weights, const SimpleTensor< int32_t > &biases, const TensorShape &dst_shape, const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const QuantizationInfo &out_quant_info) |
template<> | |
SimpleTensor< uint8_t > | depthwise_convolution (const SimpleTensor< uint8_t > &src, const SimpleTensor< int8_t > &weights, const SimpleTensor< int32_t > &biases, const TensorShape &dst_shape, const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const QuantizationInfo &out_quant_info) |
template<> | |
SimpleTensor< int8_t > | depthwise_convolution (const SimpleTensor< int8_t > &src, const SimpleTensor< int8_t > &weights, const SimpleTensor< int32_t > &biases, const TensorShape &dst_shape, const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation, const QuantizationInfo &out_quant_info) |
template<typename T , typename TW , typename TB > | |
SimpleTensor< T > | depthwise_convolution (const SimpleTensor< T > &src, const SimpleTensor< TW > &weights, const SimpleTensor< TB > &biases, const TensorShape &dst_shape, const PadStrideInfo &conv_info, unsigned int depth_multiplier, const Size2D &dilation=Size2D(1U, 1U), const QuantizationInfo &out_quant_info=QuantizationInfo(0.0f, 0)) |
template<typename T > | |
SimpleTensor< T > | depthwise_separable_convolution_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &depthwise_weights, const SimpleTensor< T > &depthwise_biases, const TensorShape &depthwise_out_shape, const SimpleTensor< T > &pointwise_weights, const SimpleTensor< T > &pointwise_biases, const TensorShape &dst_shape, const PadStrideInfo &depthwise_conv_info, const PadStrideInfo &pointwise_conv_info) |
template<typename TOut , typename TIn > | |
SimpleTensor< TOut > | dequantization_layer (const SimpleTensor< TIn > &src) |
template SimpleTensor< half > | dequantization_layer (const SimpleTensor< uint8_t > &src) |
template SimpleTensor< half > | dequantization_layer (const SimpleTensor< int8_t > &src) |
template SimpleTensor< half > | dequantization_layer (const SimpleTensor< int16_t > &src) |
template<typename T , typename U > | |
std::pair< SimpleTensor< T >, SimpleTensor< T > > | derivative (const SimpleTensor< U > &src, BorderMode border_mode, uint8_t constant_border_value, GradientDimension gradient_dimension) |
template std::pair< SimpleTensor< int16_t >, SimpleTensor< int16_t > > | derivative (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value, GradientDimension gradient_dimension) |
template<typename T > | |
SimpleTensor< T > | rdft_1d (const SimpleTensor< T > &src) |
Performs an one dimensional DFT on a real input. More... | |
template<typename T > | |
SimpleTensor< T > | ridft_1d (const SimpleTensor< T > &src, bool is_odd=false) |
Performs an one dimensional inverse DFT on a real input. More... | |
template<typename T > | |
SimpleTensor< T > | dft_1d (const SimpleTensor< T > &src, FFTDirection direction) |
Performs an one dimensional DFT on a complex input. More... | |
template<typename T > | |
SimpleTensor< T > | rdft_2d (const SimpleTensor< T > &src) |
Performs a two dimensional DFT on a real input. More... | |
template<typename T > | |
SimpleTensor< T > | ridft_2d (const SimpleTensor< T > &src, bool is_odd=false) |
Performs a two dimensional inverse DFT on a real input. More... | |
template<typename T > | |
SimpleTensor< T > | dft_2d (const SimpleTensor< T > &src, FFTDirection direction) |
Performs a two dimensional DFT on a complex input. More... | |
template<typename T > | |
SimpleTensor< T > | conv2d_dft (const SimpleTensor< T > &src, const SimpleTensor< T > &w, const PadStrideInfo &conv_info) |
Performs and DFT based convolution on a real input. More... | |
template SimpleTensor< float > | rdft_1d (const SimpleTensor< float > &src) |
template SimpleTensor< float > | ridft_1d (const SimpleTensor< float > &src, bool is_odd) |
template SimpleTensor< float > | dft_1d (const SimpleTensor< float > &src, FFTDirection direction) |
template SimpleTensor< float > | rdft_2d (const SimpleTensor< float > &src) |
template SimpleTensor< float > | ridft_2d (const SimpleTensor< float > &src, bool is_odd) |
template SimpleTensor< float > | dft_2d (const SimpleTensor< float > &src, FFTDirection direction) |
template SimpleTensor< float > | conv2d_dft (const SimpleTensor< float > &src, const SimpleTensor< float > &w, const PadStrideInfo &conv_info) |
template SimpleTensor< half > | rdft_1d (const SimpleTensor< half > &src) |
template SimpleTensor< half > | ridft_1d (const SimpleTensor< half > &src, bool is_odd) |
template SimpleTensor< half > | dft_1d (const SimpleTensor< half > &src, FFTDirection direction) |
template SimpleTensor< half > | rdft_2d (const SimpleTensor< half > &src) |
template SimpleTensor< half > | ridft_2d (const SimpleTensor< half > &src, bool is_odd) |
template SimpleTensor< half > | dft_2d (const SimpleTensor< half > &src, FFTDirection direction) |
template SimpleTensor< half > | conv2d_dft (const SimpleTensor< half > &src, const SimpleTensor< half > &w, const PadStrideInfo &conv_info) |
template<typename T > | |
SimpleTensor< T > | dilate (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | dilate (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | elementwise_unary (const SimpleTensor< T > &src, ElementWiseUnary op) |
template SimpleTensor< float > | elementwise_unary (const SimpleTensor< float > &src, ElementWiseUnary op) |
template SimpleTensor< half > | elementwise_unary (const SimpleTensor< half > &src, ElementWiseUnary op) |
template SimpleTensor< int32_t > | elementwise_unary (const SimpleTensor< int32_t > &src, ElementWiseUnary op) |
template<typename T > | |
SimpleTensor< T > | equalize_histogram (const SimpleTensor< T > &src) |
template SimpleTensor< uint8_t > | equalize_histogram (const SimpleTensor< uint8_t > &src) |
template<typename T > | |
SimpleTensor< T > | erode (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | erode (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
std::vector< KeyPoint > | fast_corners (const SimpleTensor< T > &src, float input_thresh, bool suppress_nonmax, BorderMode border_mode, T constant_border_value) |
template std::vector< KeyPoint > | fast_corners (const SimpleTensor< uint8_t > &src, float threshold, bool suppress_nonmax, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | flatten_layer (const SimpleTensor< T > &src, const TensorShape &shape_flatten) |
template SimpleTensor< float > | flatten_layer (const SimpleTensor< float > &src, const TensorShape &shape_flatten) |
template SimpleTensor< half > | flatten_layer (const SimpleTensor< half > &src, const TensorShape &shape_flatten) |
template<typename T > | |
SimpleTensor< T > | floor_layer (const SimpleTensor< T > &src) |
template SimpleTensor< half > | floor_layer (const SimpleTensor< half > &src) |
template SimpleTensor< float > | floor_layer (const SimpleTensor< float > &src) |
template<typename T , typename TB > | |
SimpleTensor< T > | fully_connected_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &weights, const SimpleTensor< TB > &bias, const TensorShape &dst_shape, QuantizationInfo out_quant_info) |
template SimpleTensor< float > | fully_connected_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &weights, const SimpleTensor< float > &bias, const TensorShape &dst_shape, QuantizationInfo out_quant_info) |
template SimpleTensor< half > | fully_connected_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &weights, const SimpleTensor< half > &bias, const TensorShape &dst_shape, QuantizationInfo out_quant_info) |
template SimpleTensor< uint8_t > | fully_connected_layer (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &dst_shape, QuantizationInfo out_quant_info) |
template SimpleTensor< int8_t > | fully_connected_layer (const SimpleTensor< int8_t > &src, const SimpleTensor< int8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &dst_shape, QuantizationInfo out_quant_info) |
template<typename T > | |
void | fuse_batch_normalization_dwc_layer (const SimpleTensor< T > &w, const SimpleTensor< T > &mean, const SimpleTensor< T > &var, SimpleTensor< T > &w_fused, SimpleTensor< T > &b_fused, const SimpleTensor< T > &b, const SimpleTensor< T > &beta, const SimpleTensor< T > &gamma, float epsilon) |
template<typename T > | |
void | fuse_batch_normalization_conv_layer (const SimpleTensor< T > &w, const SimpleTensor< T > &mean, const SimpleTensor< T > &var, SimpleTensor< T > &w_fused, SimpleTensor< T > &b_fused, const SimpleTensor< T > &b, const SimpleTensor< T > &beta, const SimpleTensor< T > &gamma, float epsilon) |
template void | fuse_batch_normalization_dwc_layer (const SimpleTensor< float > &w, const SimpleTensor< float > &mean, const SimpleTensor< float > &var, SimpleTensor< float > &w_fused, SimpleTensor< float > &b_fused, const SimpleTensor< float > &b, const SimpleTensor< float > &beta, const SimpleTensor< float > &gamma, float epsilon) |
template void | fuse_batch_normalization_dwc_layer (const SimpleTensor< half > &w, const SimpleTensor< half > &mean, const SimpleTensor< half > &var, SimpleTensor< half > &w_fused, SimpleTensor< half > &b_fused, const SimpleTensor< half > &b, const SimpleTensor< half > &beta, const SimpleTensor< half > &gamma, float epsilon) |
template void | fuse_batch_normalization_conv_layer (const SimpleTensor< float > &w, const SimpleTensor< float > &mean, const SimpleTensor< float > &var, SimpleTensor< float > &w_fused, SimpleTensor< float > &b_fused, const SimpleTensor< float > &b, const SimpleTensor< float > &beta, const SimpleTensor< float > &gamma, float epsilon) |
template void | fuse_batch_normalization_conv_layer (const SimpleTensor< half > &w, const SimpleTensor< half > &mean, const SimpleTensor< half > &var, SimpleTensor< half > &w_fused, SimpleTensor< half > &b_fused, const SimpleTensor< half > &b, const SimpleTensor< half > &beta, const SimpleTensor< half > &gamma, float epsilon) |
template<typename T > | |
SimpleTensor< T > | gather (const SimpleTensor< T > &src, const SimpleTensor< uint32_t > &indices, uint32_t actual_axis) |
template SimpleTensor< float > | gather (const SimpleTensor< float > &src, const SimpleTensor< uint32_t > &indices, uint32_t actual_axis) |
template SimpleTensor< half > | gather (const SimpleTensor< half > &src, const SimpleTensor< uint32_t > &indices, uint32_t actual_axis) |
template SimpleTensor< uint16_t > | gather (const SimpleTensor< uint16_t > &src, const SimpleTensor< uint32_t > &indices, uint32_t actual_axis) |
template SimpleTensor< uint8_t > | gather (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint32_t > &indices, uint32_t actual_axis) |
template<typename T > | |
SimpleTensor< T > | gaussian3x3 (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | gaussian3x3 (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | gaussian5x5 (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | gaussian5x5 (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
std::vector< SimpleTensor< T > > | gaussian_pyramid_half (const SimpleTensor< T > &src, BorderMode border_mode, uint8_t constant_border_value, size_t num_levels) |
template std::vector< SimpleTensor< uint8_t > > | gaussian_pyramid_half (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value, size_t num_levels) |
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > | |
SimpleTensor< T > | gemm (const SimpleTensor< T > &a, const SimpleTensor< T > &b, const SimpleTensor< T > &c, float alpha, float beta) |
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > | |
SimpleTensor< T > | gemm_mixed_precision (const SimpleTensor< T > &a, const SimpleTensor< T > &b, const SimpleTensor< T > &c, float alpha, float beta) |
template SimpleTensor< float > | gemm (const SimpleTensor< float > &a, const SimpleTensor< float > &b, const SimpleTensor< float > &c, float alpha, float beta) |
template SimpleTensor< half > | gemm (const SimpleTensor< half > &a, const SimpleTensor< half > &b, const SimpleTensor< half > &c, float alpha, float beta) |
template SimpleTensor< half > | gemm_mixed_precision (const SimpleTensor< half > &a, const SimpleTensor< half > &b, const SimpleTensor< half > &c, float alpha, float beta) |
template<typename T > | |
SimpleTensor< T > | gemm_interleave_4x4 (const SimpleTensor< T > &in, SimpleTensor< T > &out) |
template<typename T > | |
T | safe_read (const SimpleTensor< T > &t, int y, int x) |
template<typename T > | |
SimpleTensor< T > | gemm_interleave_blocked (const SimpleTensor< T > &in, SimpleTensor< T > &out, int int_by, int block, bool transposed) |
template SimpleTensor< uint8_t > | gemm_interleave_blocked (const SimpleTensor< uint8_t > &in, SimpleTensor< uint8_t > &out, int int_by, int block, bool transposed) |
template<typename T_out , typename T_in , typename T_in_1 > | |
SimpleTensor< T_out > | gemmlowp_matrix_multiply_core (const SimpleTensor< T_in > &a, const SimpleTensor< T_in_1 > &b, TensorShape shape_c, int32_t a_offset, int32_t b_offset) |
template<typename T1 , typename T2 , typename T3 > | |
SimpleTensor< T1 > | gemmlowp (const SimpleTensor< T2 > &a, const SimpleTensor< T3 > &b, TensorShape shape_c) |
template<typename TIn , typename TOut > | |
SimpleTensor< TOut > | gemmlowp_quantize_down_scale (const SimpleTensor< TIn > &in, int32_t result_offset, std::vector< int32_t > result_mult_int, std::vector< int32_t > result_shift, int32_t min, int32_t max) |
template<typename TIn , typename TOut > | |
SimpleTensor< TOut > | gemmlowp_quantize_down_scale (const SimpleTensor< TIn > &in, const SimpleTensor< TIn > &bias, int32_t result_offset, std::vector< int32_t > result_mult_int, std::vector< int32_t > result_shift, int32_t min, int32_t max) |
template<typename TIn , typename TOut > | |
SimpleTensor< TOut > | gemmlowp_quantize_down_scale_by_fixedpoint (const SimpleTensor< TIn > &in, std::vector< int32_t > result_fixedpoint_multiplier, std::vector< int32_t > result_shift, int32_t result_offset_after_shift, int32_t min, int32_t max) |
template<typename TIn , typename TOut > | |
SimpleTensor< TOut > | gemmlowp_quantize_down_scale_by_fixedpoint (const SimpleTensor< TIn > &in, const SimpleTensor< TIn > &bias, std::vector< int32_t > result_fixedpoint_multiplier, std::vector< int32_t > result_shift, int32_t result_offset_after_shift, int32_t min, int32_t max) |
template<typename TIn , typename TOut > | |
SimpleTensor< TOut > | gemmlowp_quantize_down_scale_by_float (const SimpleTensor< TIn > &in, const SimpleTensor< TIn > &bias, std::vector< float_t > result_real_multiplier, int32_t result_offset, int32_t min, int32_t max) |
template<typename TIn , typename TOut > | |
SimpleTensor< TOut > | gemmlowp_quantize_down_scale_by_float (const SimpleTensor< TIn > &in, std::vector< float_t > result_real_multiplier, int32_t result_offset, int32_t min, int32_t max) |
template SimpleTensor< uint8_t > | gemmlowp_quantize_down_scale_by_float (const SimpleTensor< int32_t > &a, const SimpleTensor< int32_t > &b, std::vector< float_t > result_real_multiplier, int32_t result_offset, int32_t min, int32_t max) |
template SimpleTensor< uint8_t > | gemmlowp_quantize_down_scale_by_float (const SimpleTensor< int32_t > &a, std::vector< float_t > result_real_multiplier, int32_t result_offset, int32_t min, int32_t max) |
template SimpleTensor< uint8_t > | gemmlowp_quantize_down_scale_by_fixedpoint (const SimpleTensor< int32_t > &a, std::vector< int32_t > result_fixedpoint_multiplier, std::vector< int32_t > result_shift, int32_t result_offset_after_shift, int32_t min, int32_t max) |
template SimpleTensor< uint8_t > | gemmlowp_quantize_down_scale_by_fixedpoint (const SimpleTensor< int32_t > &a, const SimpleTensor< int32_t > &b, std::vector< int32_t > result_fixedpoint_multiplier, std::vector< int32_t > result_shift, int32_t result_offset_after_shift, int32_t min, int32_t max) |
template SimpleTensor< uint8_t > | gemmlowp_quantize_down_scale (const SimpleTensor< int32_t > &a, int32_t result_offset, std::vector< int32_t > result_mult_int, std::vector< int32_t > result_shift, int32_t min, int32_t max) |
template SimpleTensor< uint8_t > | gemmlowp_quantize_down_scale (const SimpleTensor< int32_t > &a, const SimpleTensor< int32_t > &b, int32_t result_offset, std::vector< int32_t > result_mult_int, std::vector< int32_t > result_shift, int32_t min, int32_t max) |
template SimpleTensor< int32_t > | gemmlowp_matrix_multiply_core (const SimpleTensor< int8_t > &a, const SimpleTensor< int8_t > &b, TensorShape shape_c, int32_t a_offset, int32_t b_offset) |
template SimpleTensor< int32_t > | gemmlowp_matrix_multiply_core (const SimpleTensor< uint8_t > &a, const SimpleTensor< uint8_t > &b, TensorShape shape_c, int32_t a_offset, int32_t b_offset) |
template SimpleTensor< int32_t > | gemmlowp< int32_t, int8_t, int8_t > (const SimpleTensor< int8_t > &a, const SimpleTensor< int8_t > &b, TensorShape shape_c) |
template SimpleTensor< int32_t > | gemmlowp< int32_t, uint8_t, uint8_t > (const SimpleTensor< uint8_t > &a, const SimpleTensor< uint8_t > &b, TensorShape shape_c) |
template SimpleTensor< int32_t > | gemmlowp< int32_t, uint8_t, int8_t > (const SimpleTensor< uint8_t > &a, const SimpleTensor< int8_t > &b, TensorShape shape_c) |
template<typename T1 , typename T2 , typename T3 > | |
SimpleTensor< T1 > | gemmlowp_matrix_multiply_core (const SimpleTensor< T2 > &a, const SimpleTensor< T3 > &b, TensorShape shape_c, int32_t a_offset, int32_t b_offset) |
template<typename TIn , typename TOut > | |
SimpleTensor< uint8_t > | gemmlowp_quantize_down_scale (const SimpleTensor< TIn > &in, int32_t result_offset, std::vector< int32_t > result_mult_int, std::vector< int32_t > result_shift) |
template<typename T > | |
SimpleTensor< T > | gemm_reshape_lhs_matrix (const SimpleTensor< T > &in, const TensorShape &output_shape, const GEMMLHSMatrixInfo &lhs_info) |
template SimpleTensor< int > | gemm_reshape_lhs_matrix (const SimpleTensor< int > &in, const TensorShape &output_shape, const GEMMLHSMatrixInfo &lhs_info) |
template SimpleTensor< short > | gemm_reshape_lhs_matrix (const SimpleTensor< short > &in, const TensorShape &output_shape, const GEMMLHSMatrixInfo &lhs_info) |
template SimpleTensor< char > | gemm_reshape_lhs_matrix (const SimpleTensor< char > &in, const TensorShape &output_shape, const GEMMLHSMatrixInfo &lhs_info) |
template<typename T > | |
SimpleTensor< T > | gemm_reshape_rhs_matrix (const SimpleTensor< T > &in, const TensorShape &output_shape, const GEMMRHSMatrixInfo &rhs_info) |
template SimpleTensor< int > | gemm_reshape_rhs_matrix (const SimpleTensor< int > &in, const TensorShape &output_shape, const GEMMRHSMatrixInfo &rhs_info) |
template SimpleTensor< short > | gemm_reshape_rhs_matrix (const SimpleTensor< short > &in, const TensorShape &output_shape, const GEMMRHSMatrixInfo &rhs_info) |
template SimpleTensor< char > | gemm_reshape_rhs_matrix (const SimpleTensor< char > &in, const TensorShape &output_shape, const GEMMRHSMatrixInfo &rhs_info) |
template<typename T > | |
SimpleTensor< T > | gemm_transpose_1xW (const SimpleTensor< T > &in) |
template<typename T > | |
std::vector< KeyPoint > | harris_corner_detector (const SimpleTensor< T > &src, float threshold, float min_dist, float sensitivity, int gradient_size, int block_size, BorderMode border_mode, T constant_border_value) |
template std::vector< KeyPoint > | harris_corner_detector (const SimpleTensor< uint8_t > &src, float threshold, float min_dist, float sensitivity, int gradient_size, int block_size, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< uint32_t > | histogram (const SimpleTensor< T > &src, size_t num_bins, int32_t offset, uint32_t range) |
template SimpleTensor< uint32_t > | histogram (const SimpleTensor< uint8_t > &src, size_t num_bins, int32_t offset, uint32_t range) |
template<typename T , typename U , typename V > | |
void | hog_orientation_binning (const SimpleTensor< T > &mag, const SimpleTensor< U > &phase, SimpleTensor< V > &hog_space, const HOGInfo &hog_info) |
template<typename T > | |
void | hog_block_normalization (SimpleTensor< T > &desc, const SimpleTensor< T > &hog_space, const HOGInfo &hog_info) |
template<typename T , typename U > | |
SimpleTensor< T > | hog_descriptor (const SimpleTensor< U > &src, BorderMode border_mode, U constant_border_value, const HOGInfo &hog_info) |
template void | hog_orientation_binning (const SimpleTensor< int16_t > &mag, const SimpleTensor< uint8_t > &phase, SimpleTensor< float > &hog_space, const HOGInfo &hog_info) |
template void | hog_block_normalization (SimpleTensor< float > &desc, const SimpleTensor< float > &hog_space, const HOGInfo &hog_info) |
template SimpleTensor< float > | hog_descriptor (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value, const HOGInfo &hog_info) |
template<typename T > | |
std::vector< DetectionWindow > | hog_detector (const SimpleTensor< T > &src, const std::vector< T > &descriptor, unsigned int max_num_detection_windows, const HOGInfo &hog_info, const Size2D &detection_window_stride, float threshold, uint16_t idx_class) |
template std::vector< DetectionWindow > | hog_detector (const SimpleTensor< float > &src, const std::vector< float > &descriptor, unsigned int max_num_detection_windows, const HOGInfo &hog_info, const Size2D &detection_window_stride, float threshold, uint16_t idx_class) |
void | detection_windows_non_maxima_suppression (std::vector< DetectionWindow > &multi_windows, float min_distance) |
template<typename T > | |
std::vector< DetectionWindow > | hog_multi_detection (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value, const std::vector< HOGInfo > &models, std::vector< std::vector< float >> descriptors, unsigned int max_num_detection_windows, float threshold, bool non_maxima_suppression, float min_distance) |
template std::vector< DetectionWindow > | hog_multi_detection (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value, const std::vector< HOGInfo > &models, std::vector< std::vector< float >> descriptors, unsigned int max_num_detection_windows, float threshold, bool non_maxima_suppression, float min_distance) |
template<typename T > | |
void | im2col_nchw (const SimpleTensor< T > &src, SimpleTensor< T > &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, unsigned int num_groups) |
template<typename T > | |
void | im2col_nhwc (const SimpleTensor< T > &src, SimpleTensor< T > &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias) |
template<typename T > | |
void | im2col (const SimpleTensor< T > &src, SimpleTensor< T > &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, unsigned int num_groups) |
template void | im2col (const SimpleTensor< uint8_t > &src, SimpleTensor< uint8_t > &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, unsigned int num_groups) |
template void | im2col (const SimpleTensor< half > &src, SimpleTensor< half > &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, unsigned int num_groups) |
template void | im2col (const SimpleTensor< float > &src, SimpleTensor< float > &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, unsigned int num_groups) |
template<typename T > | |
SimpleTensor< T > | instance_normalization (const SimpleTensor< T > &src, float gamma, float beta, float epsilon) |
template SimpleTensor< float > | instance_normalization (const SimpleTensor< float > &src, float gamma, float beta, float epsilon) |
template SimpleTensor< half > | instance_normalization (const SimpleTensor< half > &src, float gamma, float beta, float epsilon) |
template<typename T > | |
SimpleTensor< uint32_t > | integral_image (const SimpleTensor< T > &src) |
template SimpleTensor< uint32_t > | integral_image (const SimpleTensor< uint8_t > &src) |
template<typename T > | |
SimpleTensor< T > | l2_normalize (const SimpleTensor< T > &src, unsigned int axis, float epsilon) |
template SimpleTensor< float > | l2_normalize (const SimpleTensor< float > &src, unsigned int axis, float epsilon) |
template SimpleTensor< half > | l2_normalize (const SimpleTensor< half > &src, unsigned int axis, float epsilon) |
template<typename T , typename U > | |
std::vector< SimpleTensor< U > > | laplacian_pyramid (const SimpleTensor< T > &src, SimpleTensor< U > &dst, size_t num_levels, BorderMode border_mode, uint8_t constant_border_value) |
template std::vector< SimpleTensor< int16_t > > | laplacian_pyramid (const SimpleTensor< uint8_t > &src, SimpleTensor< int16_t > &dst, size_t num_levels, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T , typename U > | |
SimpleTensor< U > | laplacian_reconstruct (const std::vector< SimpleTensor< T >> &pyramid, const SimpleTensor< T > &low_res, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | laplacian_reconstruct (const std::vector< SimpleTensor< int16_t >> &pyramid, const SimpleTensor< int16_t > &low_res, BorderMode border_mode, int16_t constant_border_value) |
template<typename T > | |
T | logical_binary_op (arm_compute::LogicalOperation op, T src1, T src2) |
template<typename T > | |
SimpleTensor< T > | logical_or (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2) |
template<typename T > | |
SimpleTensor< T > | logical_and (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2) |
template<typename T > | |
SimpleTensor< T > | logical_not (const SimpleTensor< T > &src) |
template SimpleTensor< uint8_t > | logical_or (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2) |
template SimpleTensor< uint8_t > | logical_and (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2) |
template SimpleTensor< uint8_t > | logical_not (const SimpleTensor< uint8_t > &src1) |
template<typename T > | |
SimpleTensor< T > | magnitude (const SimpleTensor< T > &gx, const SimpleTensor< T > &gy, MagnitudeType magnitude_type) |
template SimpleTensor< int16_t > | magnitude (const SimpleTensor< int16_t > &gx, const SimpleTensor< int16_t > &gy, MagnitudeType magnitude_type) |
template SimpleTensor< int32_t > | magnitude (const SimpleTensor< int32_t > &gx, const SimpleTensor< int32_t > &gy, MagnitudeType magnitude_type) |
template SimpleTensor< half_float::half > | magnitude (const SimpleTensor< half_float::half > &gx, const SimpleTensor< half_float::half > &gy, MagnitudeType magnitude_type) |
template<typename T > | |
SimpleTensor< T > | max_unpooling_layer_internal (const SimpleTensor< T > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > &indices, TensorShape output_shape, DataLayout data_layout) |
template<> | |
SimpleTensor< uint8_t > | max_unpooling_layer< uint8_t > (const SimpleTensor< uint8_t > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > &indices, TensorShape output_shape, DataLayout data_layout) |
template<typename T > | |
SimpleTensor< T > | max_unpooling_layer (const SimpleTensor< T > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > &indices, TensorShape output_shape, DataLayout data_layout) |
template SimpleTensor< float > | max_unpooling_layer (const SimpleTensor< float > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > &indices, TensorShape output_shape, DataLayout data_layout) |
template SimpleTensor< half > | max_unpooling_layer (const SimpleTensor< half > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > &indices, TensorShape output_shape, DataLayout data_layout) |
template<typename T > | |
std::pair< float, float > | mean_and_standard_deviation (const SimpleTensor< T > &in) |
template std::pair< float, float > | mean_and_standard_deviation (const SimpleTensor< uint8_t > &in) |
template std::pair< float, float > | mean_and_standard_deviation (const SimpleTensor< half > &in) |
template std::pair< float, float > | mean_and_standard_deviation (const SimpleTensor< float > &in) |
template<typename T > | |
SimpleTensor< T > | mean_std_normalization_layer (const SimpleTensor< T > &src, float epsilon) |
template SimpleTensor< float > | mean_std_normalization_layer (const SimpleTensor< float > &src, float epsilon) |
template SimpleTensor< half > | mean_std_normalization_layer (const SimpleTensor< half > &src, float epsilon) |
template<typename T > | |
SimpleTensor< T > | median3x3 (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | median3x3 (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
void | compute_min_max (const SimpleTensor< T > &src, T &min, T &max) |
template<typename T > | |
MinMaxLocationValues< T > | min_max_location (const SimpleTensor< T > &src) |
template MinMaxLocationValues< uint8_t > | min_max_location (const SimpleTensor< uint8_t > &src) |
template MinMaxLocationValues< int16_t > | min_max_location (const SimpleTensor< int16_t > &src) |
template MinMaxLocationValues< float > | min_max_location (const SimpleTensor< float > &src) |
template<typename T > | |
SimpleTensor< T > | non_linear_filter (const SimpleTensor< T > &src, NonLinearFilterFunction function, unsigned int mask_size, MatrixPattern pattern, const uint8_t *mask, BorderMode border_mode, uint8_t constant_border_value) |
template SimpleTensor< uint8_t > | non_linear_filter (const SimpleTensor< uint8_t > &src, NonLinearFilterFunction function, unsigned int mask_size, MatrixPattern pattern, const uint8_t *mask, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | non_maxima_suppression (const SimpleTensor< T > &src, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< float > | non_maxima_suppression (const SimpleTensor< float > &src, BorderMode border_mode, float constant_border_value) |
template SimpleTensor< uint8_t > | non_maxima_suppression (const SimpleTensor< uint8_t > &src, BorderMode border_mode, uint8_t constant_border_value) |
SimpleTensor< int > | non_max_suppression (const SimpleTensor< float > &bboxes, const SimpleTensor< float > &scores, SimpleTensor< int > &indices, unsigned int max_output_size, float score_threshold, float nms_threshold) |
template<typename T > | |
SimpleTensor< T > | normalization_layer (const SimpleTensor< T > &src, NormalizationLayerInfo info) |
template SimpleTensor< float > | normalization_layer (const SimpleTensor< float > &src, NormalizationLayerInfo info) |
template SimpleTensor< half > | normalization_layer (const SimpleTensor< half > &src, NormalizationLayerInfo info) |
template<typename T > | |
SimpleTensor< T > | normalize_planar_yuv_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &mean, const SimpleTensor< T > &std) |
template<> | |
SimpleTensor< uint8_t > | normalize_planar_yuv_layer< uint8_t > (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &mean, const SimpleTensor< uint8_t > &std) |
template<> | |
SimpleTensor< int8_t > | normalize_planar_yuv_layer< int8_t > (const SimpleTensor< int8_t > &src, const SimpleTensor< int8_t > &mean, const SimpleTensor< int8_t > &std) |
template SimpleTensor< half > | normalize_planar_yuv_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &mean, const SimpleTensor< half > &std) |
template SimpleTensor< float > | normalize_planar_yuv_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &mean, const SimpleTensor< float > &std) |
template<typename T > | |
std::vector< KeyPoint > | optical_flow (const SimpleTensor< T > &old_input, const SimpleTensor< T > &new_input, const OpticalFlowParameters ¶ms, size_t num_levels, const std::vector< KeyPoint > &old_points, const std::vector< KeyPoint > &new_points_estimates, BorderMode border_mode, uint8_t constant_border_value) |
template std::vector< KeyPoint > | optical_flow (const SimpleTensor< uint8_t > &old_input, const SimpleTensor< uint8_t > &new_input, const OpticalFlowParameters ¶ms, size_t num_levels, const std::vector< KeyPoint > &old_points, const std::vector< KeyPoint > &new_points_estimates, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | pad_layer (const SimpleTensor< T > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode=PaddingMode::CONSTANT) |
Reference function to pad an ND tensor. More... | |
template SimpleTensor< float > | pad_layer (const SimpleTensor< float > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template SimpleTensor< half > | pad_layer (const SimpleTensor< half > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template SimpleTensor< uint8_t > | pad_layer (const SimpleTensor< uint8_t > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template SimpleTensor< int8_t > | pad_layer (const SimpleTensor< int8_t > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template SimpleTensor< uint16_t > | pad_layer (const SimpleTensor< uint16_t > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template SimpleTensor< int16_t > | pad_layer (const SimpleTensor< int16_t > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template SimpleTensor< uint32_t > | pad_layer (const SimpleTensor< uint32_t > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template SimpleTensor< int32_t > | pad_layer (const SimpleTensor< int32_t > &src, const PaddingList &paddings, const PixelValue const_value=PixelValue(), const PaddingMode mode) |
template<typename T > | |
SimpleTensor< T > | permute (const SimpleTensor< T > &src, PermutationVector perm) |
template SimpleTensor< int8_t > | permute (const SimpleTensor< int8_t > &src, PermutationVector perm) |
template SimpleTensor< uint8_t > | permute (const SimpleTensor< uint8_t > &src, PermutationVector perm) |
template SimpleTensor< int16_t > | permute (const SimpleTensor< int16_t > &src, PermutationVector perm) |
template SimpleTensor< uint16_t > | permute (const SimpleTensor< uint16_t > &src, PermutationVector perm) |
template SimpleTensor< uint32_t > | permute (const SimpleTensor< uint32_t > &src, PermutationVector perm) |
template SimpleTensor< float > | permute (const SimpleTensor< float > &src, PermutationVector perm) |
template SimpleTensor< half > | permute (const SimpleTensor< half > &src, PermutationVector perm) |
template<typename T > | |
SimpleTensor< uint8_t > | phase (const SimpleTensor< T > &gx, const SimpleTensor< T > &gy, PhaseType phase_type) |
template SimpleTensor< uint8_t > | phase (const SimpleTensor< int16_t > &gx, const SimpleTensor< int16_t > &gy, PhaseType phase_type) |
template SimpleTensor< uint8_t > | phase (const SimpleTensor< int32_t > &gx, const SimpleTensor< int32_t > &gy, PhaseType phase_type) |
template<typename T1 , typename T2 , typename T3 > | |
SimpleTensor< T3 > | pixel_wise_multiplication (const SimpleTensor< T1 > &src1, const SimpleTensor< T2 > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template<> | |
SimpleTensor< uint8_t > | pixel_wise_multiplication (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template<> | |
SimpleTensor< int16_t > | pixel_wise_multiplication (const SimpleTensor< uint8_t > &src1, const SimpleTensor< uint8_t > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template<> | |
SimpleTensor< int8_t > | pixel_wise_multiplication (const SimpleTensor< int8_t > &src1, const SimpleTensor< int8_t > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template<> | |
SimpleTensor< int16_t > | pixel_wise_multiplication (const SimpleTensor< int16_t > &src1, const SimpleTensor< int16_t > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template SimpleTensor< int16_t > | pixel_wise_multiplication (const SimpleTensor< uint8_t > &src1, const SimpleTensor< int16_t > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template SimpleTensor< int32_t > | pixel_wise_multiplication (const SimpleTensor< int32_t > &src1, const SimpleTensor< int32_t > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template SimpleTensor< float > | pixel_wise_multiplication (const SimpleTensor< float > &src1, const SimpleTensor< float > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template SimpleTensor< half_float::half > | pixel_wise_multiplication (const SimpleTensor< half_float::half > &src1, const SimpleTensor< half_float::half > &src2, float scale, ConvertPolicy convert_policy, RoundingPolicy rounding_policy, DataType dt_out, const QuantizationInfo &qout) |
template<typename T , typename ACC_T , typename std::enable_if< is_floating_point< T >::value, int >::type > | |
SimpleTensor< T > | pooling_layer_internal (const SimpleTensor< T > &src, const PoolingLayerInfo &info, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template SimpleTensor< float > | pooling_layer_internal< float > (const SimpleTensor< float > &src, const PoolingLayerInfo &info, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template SimpleTensor< half > | pooling_layer_internal< half > (const SimpleTensor< half > &src, const PoolingLayerInfo &info, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template SimpleTensor< half > | pooling_layer_internal< half, float > (const SimpleTensor< half > &src, const PoolingLayerInfo &info, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template<typename T > | |
SimpleTensor< T > | pooling_layer (const SimpleTensor< T > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template<> | |
SimpleTensor< uint8_t > | pooling_layer< uint8_t > (const SimpleTensor< uint8_t > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template<> | |
SimpleTensor< int8_t > | pooling_layer< int8_t > (const SimpleTensor< int8_t > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template<> | |
SimpleTensor< half > | pooling_layer (const SimpleTensor< half > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template SimpleTensor< float > | pooling_layer (const SimpleTensor< float > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > *indices, DataLayout data_layout) |
template<typename T > | |
SimpleTensor< T > | prior_box_layer (const SimpleTensor< T > &src1, const SimpleTensor< T > &src2, const PriorBoxLayerInfo &info, const TensorShape &output_shape) |
template SimpleTensor< float > | prior_box_layer (const SimpleTensor< float > &src1, const SimpleTensor< float > &src2, const PriorBoxLayerInfo &info, const TensorShape &output_shape) |
SimpleTensor< int16_t > | qlstm_layer_normalization (const SimpleTensor< int16_t > &src, const SimpleTensor< int16_t > &weight, const SimpleTensor< int32_t > &bias) |
template<typename Tin , typename Tout > | |
SimpleTensor< Tout > | quantization_layer (const SimpleTensor< Tin > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template<> | |
SimpleTensor< uint8_t > | quantization_layer (const SimpleTensor< uint8_t > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template<> | |
SimpleTensor< int8_t > | quantization_layer (const SimpleTensor< uint8_t > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template<> | |
SimpleTensor< uint8_t > | quantization_layer (const SimpleTensor< int8_t > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template<> | |
SimpleTensor< int8_t > | quantization_layer (const SimpleTensor< int8_t > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template<> | |
SimpleTensor< uint16_t > | quantization_layer (const SimpleTensor< uint8_t > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template SimpleTensor< int8_t > | quantization_layer (const SimpleTensor< half > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template SimpleTensor< int8_t > | quantization_layer (const SimpleTensor< float > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
template<typename T > | |
SimpleTensor< T > | range (SimpleTensor< T > &dst, float start, const size_t num_of_elements, float step) |
template<> | |
SimpleTensor< uint8_t > | range (SimpleTensor< uint8_t > &dst, float start, const size_t num_of_elements, float step) |
template SimpleTensor< float > | range (SimpleTensor< float > &dst, float start, const size_t num_of_elements, float step) |
template SimpleTensor< half > | range (SimpleTensor< half > &dst, float start, const size_t num_of_elements, float step) |
template SimpleTensor< int8_t > | range (SimpleTensor< int8_t > &dst, float start, const size_t num_of_elements, float step) |
template SimpleTensor< uint16_t > | range (SimpleTensor< uint16_t > &dst, float start, const size_t num_of_elements, float step) |
template SimpleTensor< int16_t > | range (SimpleTensor< int16_t > &dst, float start, const size_t num_of_elements, float step) |
template<typename T , typename OT > | |
SimpleTensor< OT > | compute_reduction_operation (const SimpleTensor< T > &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op) |
template<typename T , typename OT > | |
SimpleTensor< OT > | reduction_operation (const SimpleTensor< T > &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, QuantizationInfo quantization_info_output) |
template<> | |
SimpleTensor< uint8_t > | reduction_operation (const SimpleTensor< uint8_t > &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, QuantizationInfo quantization_info_output) |
template<> | |
SimpleTensor< int8_t > | reduction_operation (const SimpleTensor< int8_t > &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, QuantizationInfo quantization_info_output) |
template SimpleTensor< float > | reduction_operation (const SimpleTensor< float > &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, QuantizationInfo quantization_info_output=QuantizationInfo()) |
template SimpleTensor< half > | reduction_operation (const SimpleTensor< half > &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, QuantizationInfo quantization_info_output=QuantizationInfo()) |
template SimpleTensor< int32_t > | reduction_operation (const SimpleTensor< int32_t > &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, QuantizationInfo quantization_info_output=QuantizationInfo()) |
template<typename T > | |
SimpleTensor< T > | remap (const SimpleTensor< T > &in, SimpleTensor< float > &map_x, SimpleTensor< float > &map_y, SimpleTensor< T > &valid_mask, InterpolationPolicy policy, BorderMode border_mode, T constant_border_value) |
template SimpleTensor< uint8_t > | remap (const SimpleTensor< uint8_t > &src, SimpleTensor< float > &map_x, SimpleTensor< float > &map_y, SimpleTensor< uint8_t > &valid_mask, InterpolationPolicy policy, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | reorg_layer (const SimpleTensor< T > &src, int32_t stride) |
template SimpleTensor< int32_t > | reorg_layer (const SimpleTensor< int32_t > &src, int32_t stride) |
template SimpleTensor< int16_t > | reorg_layer (const SimpleTensor< int16_t > &src, int32_t stride) |
template SimpleTensor< int8_t > | reorg_layer (const SimpleTensor< int8_t > &src, int32_t stride) |
template<typename T > | |
SimpleTensor< T > | reshape_layer (const SimpleTensor< T > &src, const TensorShape &output_shape) |
[ReshapeLayer] More... | |
template SimpleTensor< uint8_t > | reshape_layer (const SimpleTensor< uint8_t > &src, const TensorShape &output_shape) |
template SimpleTensor< int8_t > | reshape_layer (const SimpleTensor< int8_t > &src, const TensorShape &output_shape) |
template SimpleTensor< uint16_t > | reshape_layer (const SimpleTensor< uint16_t > &src, const TensorShape &output_shape) |
template SimpleTensor< int16_t > | reshape_layer (const SimpleTensor< int16_t > &src, const TensorShape &output_shape) |
template SimpleTensor< uint32_t > | reshape_layer (const SimpleTensor< uint32_t > &src, const TensorShape &output_shape) |
template SimpleTensor< int32_t > | reshape_layer (const SimpleTensor< int32_t > &src, const TensorShape &output_shape) |
template SimpleTensor< half > | reshape_layer (const SimpleTensor< half > &src, const TensorShape &output_shape) |
template SimpleTensor< float > | reshape_layer (const SimpleTensor< float > &src, const TensorShape &output_shape) |
template<typename T > | |
SimpleTensor< T > | reverse (const SimpleTensor< T > &src, const SimpleTensor< uint32_t > &axis) |
template SimpleTensor< uint8_t > | reverse (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint32_t > &axis) |
template SimpleTensor< half > | reverse (const SimpleTensor< half > &src, const SimpleTensor< uint32_t > &axis) |
template SimpleTensor< float > | reverse (const SimpleTensor< float > &src, const SimpleTensor< uint32_t > &axis) |
template<> | |
SimpleTensor< float > | roi_align_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &rois, const ROIPoolingLayerInfo &pool_info, const QuantizationInfo &output_qinfo) |
template<> | |
SimpleTensor< half > | roi_align_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &rois, const ROIPoolingLayerInfo &pool_info, const QuantizationInfo &output_qinfo) |
template<> | |
SimpleTensor< uint8_t > | roi_align_layer (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint16_t > &rois, const ROIPoolingLayerInfo &pool_info, const QuantizationInfo &output_qinfo) |
template<> | |
SimpleTensor< int8_t > | roi_align_layer (const SimpleTensor< int8_t > &src, const SimpleTensor< uint16_t > &rois, const ROIPoolingLayerInfo &pool_info, const QuantizationInfo &output_qinfo) |
template<typename T , typename TRois > | |
SimpleTensor< T > | roi_align_layer (const SimpleTensor< T > &src, const SimpleTensor< TRois > &rois, const ROIPoolingLayerInfo &pool_info, const QuantizationInfo &output_qinfo) |
template<typename T > | |
SimpleTensor< T > | scale_core (const SimpleTensor< T > &in, float scale_x, float scale_y, InterpolationPolicy policy, BorderMode border_mode, T constant_border_value, SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) |
template<typename T > | |
SimpleTensor< T > | scale (const SimpleTensor< T > &src, float scale_x, float scale_y, InterpolationPolicy policy, BorderMode border_mode, T constant_border_value, SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) |
template<> | |
SimpleTensor< uint8_t > | scale (const SimpleTensor< uint8_t > &src, float scale_x, float scale_y, InterpolationPolicy policy, BorderMode border_mode, uint8_t constant_border_value, SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) |
template<> | |
SimpleTensor< int8_t > | scale (const SimpleTensor< int8_t > &src, float scale_x, float scale_y, InterpolationPolicy policy, BorderMode border_mode, int8_t constant_border_value, SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) |
template SimpleTensor< int16_t > | scale (const SimpleTensor< int16_t > &src, float scale_x, float scale_y, InterpolationPolicy policy, BorderMode border_mode, int16_t constant_border_value, SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) |
template SimpleTensor< half > | scale (const SimpleTensor< half > &src, float scale_x, float scale_y, InterpolationPolicy policy, BorderMode border_mode, half constant_border_value, SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) |
template SimpleTensor< float > | scale (const SimpleTensor< float > &src, float scale_x, float scale_y, InterpolationPolicy policy, BorderMode border_mode, float constant_border_value, SamplingPolicy sampling_policy, bool ceil_policy_scale, bool align_corners) |
template<typename T , typename U > | |
std::pair< SimpleTensor< T >, SimpleTensor< T > > | scharr (const SimpleTensor< U > &src, int filter_size, BorderMode border_mode, uint8_t constant_border_value, GradientDimension gradient_dimension) |
template std::pair< SimpleTensor< int16_t >, SimpleTensor< int16_t > > | scharr (const SimpleTensor< uint8_t > &src, int filter_size, BorderMode border_mode, uint8_t constant_border_value, GradientDimension gradient_dimension) |
template<typename T > | |
SimpleTensor< T > | select (const SimpleTensor< uint8_t > &c, const SimpleTensor< T > &x, const SimpleTensor< T > &y) |
template SimpleTensor< uint8_t > | select (const SimpleTensor< uint8_t > &c, const SimpleTensor< uint8_t > &x, const SimpleTensor< uint8_t > &y) |
template SimpleTensor< half > | select (const SimpleTensor< uint8_t > &c, const SimpleTensor< half > &x, const SimpleTensor< half > &y) |
template SimpleTensor< float > | select (const SimpleTensor< uint8_t > &c, const SimpleTensor< float > &x, const SimpleTensor< float > &y) |
template<typename T > | |
SimpleTensor< T > | slice (const SimpleTensor< T > &src, Coordinates starts, Coordinates ends) |
template SimpleTensor< float > | slice (const SimpleTensor< float > &src, Coordinates starts, Coordinates ends) |
template SimpleTensor< half_float::half > | slice (const SimpleTensor< half_float::half > &src, Coordinates starts, Coordinates ends) |
template<typename T > | |
SimpleTensor< T > | strided_slice (const SimpleTensor< T > &src, Coordinates starts, Coordinates ends, BiStrides strides, int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) |
template SimpleTensor< float > | strided_slice (const SimpleTensor< float > &src, Coordinates starts, Coordinates ends, BiStrides strides, int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) |
template SimpleTensor< half_float::half > | strided_slice (const SimpleTensor< half_float::half > &src, Coordinates starts, Coordinates ends, BiStrides strides, int32_t begin_mask, int32_t end_mask, int32_t shrink_axis_mask) |
template<typename T , typename U > | |
std::pair< SimpleTensor< T >, SimpleTensor< T > > | sobel (const SimpleTensor< U > &src, int filter_size, BorderMode border_mode, uint8_t constant_border_value, GradientDimension gradient_dimension) |
template std::pair< SimpleTensor< int16_t >, SimpleTensor< int16_t > > | sobel (const SimpleTensor< uint8_t > &src, int filter_size, BorderMode border_mode, uint8_t constant_border_value, GradientDimension gradient_dimension) |
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > | |
SimpleTensor< T > | softmax_layer_generic (const SimpleTensor< T > &src, float beta, int32_t axis, bool is_log) |
template SimpleTensor< float > | softmax_layer_generic (const SimpleTensor< float > &src, float beta, int32_t axis, bool is_log) |
template SimpleTensor< half > | softmax_layer_generic (const SimpleTensor< half > &src, float beta, int32_t axis, bool is_log) |
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > | |
SimpleTensor< T > | softmax_layer (const SimpleTensor< T > &src, float beta, int32_t axis, bool is_log) |
template SimpleTensor< float > | softmax_layer (const SimpleTensor< float > &src, float beta, int32_t axis, bool is_log) |
template SimpleTensor< half > | softmax_layer (const SimpleTensor< half > &src, float beta, int32_t axis, bool is_log) |
template SimpleTensor< uint8_t > | softmax_layer (const SimpleTensor< uint8_t > &src, float beta, int32_t axis, bool is_log) |
template SimpleTensor< int8_t > | softmax_layer (const SimpleTensor< int8_t > &src, float beta, int32_t axis, bool is_log) |
template<typename T > | |
SimpleTensor< T > | space_to_batch (const SimpleTensor< T > &src, const SimpleTensor< int32_t > &block_shape, const SimpleTensor< int32_t > &paddings, const TensorShape &dst_shape) |
template SimpleTensor< float > | space_to_batch (const SimpleTensor< float > &src, const SimpleTensor< int32_t > &block_shape, const SimpleTensor< int32_t > &paddings, const TensorShape &dst_shape) |
template SimpleTensor< half > | space_to_batch (const SimpleTensor< half > &src, const SimpleTensor< int32_t > &block_shape, const SimpleTensor< int32_t > &paddings, const TensorShape &dst_shape) |
template SimpleTensor< uint8_t > | space_to_batch (const SimpleTensor< uint8_t > &src, const SimpleTensor< int32_t > &block_shape, const SimpleTensor< int32_t > &paddings, const TensorShape &dst_shape) |
template<typename T > | |
SimpleTensor< T > | space_to_depth (const SimpleTensor< T > &src, const TensorShape &dst_shape, const int block_shape) |
template SimpleTensor< float > | space_to_depth (const SimpleTensor< float > &src, const TensorShape &dst_shape, const int block_shape) |
template SimpleTensor< half > | space_to_depth (const SimpleTensor< half > &src, const TensorShape &dst_shape, const int block_shape) |
template<typename T > | |
SimpleTensor< T > | stack_layer (const std::vector< SimpleTensor< T >> &in, const TensorShape &output_shape, DataType data_type, unsigned int axis) |
template SimpleTensor< int > | stack_layer (const std::vector< SimpleTensor< int >> &in, const TensorShape &output_shape, DataType data_type, unsigned int axis) |
template SimpleTensor< short > | stack_layer (const std::vector< SimpleTensor< short >> &in, const TensorShape &output_shape, DataType data_type, unsigned int axis) |
template SimpleTensor< char > | stack_layer (const std::vector< SimpleTensor< char >> &in, const TensorShape &output_shape, DataType data_type, unsigned int axis) |
template<typename T > | |
SimpleTensor< T > | table_lookup (const SimpleTensor< T > &src, const std::map< T, T > &rawlut) |
template SimpleTensor< uint8_t > | table_lookup (const SimpleTensor< uint8_t > &src, const std::map< uint8_t, uint8_t > &rawlut) |
template SimpleTensor< int16_t > | table_lookup (const SimpleTensor< int16_t > &src, const std::map< int16_t, int16_t > &rawlut) |
template<typename T > | |
SimpleTensor< T > | threshold (const SimpleTensor< T > &src, T threshold, T false_value, T true_value, ThresholdType type, T upper) |
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) |
template<typename T > | |
SimpleTensor< T > | tile (const SimpleTensor< T > &src, const Multiples &multiples) |
template SimpleTensor< uint8_t > | tile (const SimpleTensor< uint8_t > &src, const Multiples &multiples) |
template SimpleTensor< int8_t > | tile (const SimpleTensor< int8_t > &src, const Multiples &multiples) |
template SimpleTensor< uint16_t > | tile (const SimpleTensor< uint16_t > &src, const Multiples &multiples) |
template SimpleTensor< int16_t > | tile (const SimpleTensor< int16_t > &src, const Multiples &multiples) |
template SimpleTensor< uint32_t > | tile (const SimpleTensor< uint32_t > &src, const Multiples &multiples) |
template SimpleTensor< int32_t > | tile (const SimpleTensor< int32_t > &src, const Multiples &multiples) |
template SimpleTensor< half > | tile (const SimpleTensor< half > &src, const Multiples &multiples) |
template SimpleTensor< float > | tile (const SimpleTensor< float > &src, const Multiples &multiples) |
template<typename T > | |
SimpleTensor< T > | transpose (const SimpleTensor< T > &src) |
template SimpleTensor< uint8_t > | transpose (const SimpleTensor< uint8_t > &src) |
template SimpleTensor< uint16_t > | transpose (const SimpleTensor< uint16_t > &src) |
template SimpleTensor< uint32_t > | transpose (const SimpleTensor< uint32_t > &src) |
template SimpleTensor< half > | transpose (const SimpleTensor< half > &src) |
template SimpleTensor< float > | transpose (const SimpleTensor< float > &src) |
template<typename T > | |
std::vector< SimpleTensor< T > > | unstack (const SimpleTensor< T > &input_tensor, std::vector< SimpleTensor< T >> &output_tensors, int axis) |
template std::vector< SimpleTensor< float > > | unstack (const SimpleTensor< float > &input_tensor, std::vector< SimpleTensor< float >> &output_tensors, int axis) |
template std::vector< SimpleTensor< half > > | unstack (const SimpleTensor< half > &input_tensor, std::vector< SimpleTensor< half >> &output_tensors, int axis) |
template std::vector< SimpleTensor< uint8_t > > | unstack (const SimpleTensor< uint8_t > &input_tensor, std::vector< SimpleTensor< uint8_t >> &output_tensors, int axis) |
bool | valid_bilinear_policy (float xn, float yn, int width, int height, BorderMode border_mode) |
template<typename T > | |
SimpleTensor< T > | warp_affine (const SimpleTensor< T > &src, SimpleTensor< T > &valid_mask, const float *matrix, InterpolationPolicy policy, BorderMode border_mode, uint8_t constant_border_value) |
template SimpleTensor< uint8_t > | warp_affine (const SimpleTensor< uint8_t > &src, SimpleTensor< uint8_t > &valid_mask, const float *matrix, InterpolationPolicy policy, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | warp_perspective (const SimpleTensor< T > &src, SimpleTensor< T > &valid_mask, const float *matrix, InterpolationPolicy policy, BorderMode border_mode, uint8_t constant_border_value) |
template SimpleTensor< uint8_t > | warp_perspective (const SimpleTensor< uint8_t > &src, SimpleTensor< uint8_t > &valid_mask, const float *matrix, InterpolationPolicy policy, BorderMode border_mode, uint8_t constant_border_value) |
template<typename T > | |
SimpleTensor< T > | weights_reshape (const SimpleTensor< T > &src, const SimpleTensor< T > &biases, const TensorShape &dst_shape, const unsigned int num_groups) |
template SimpleTensor< float > | weights_reshape (const SimpleTensor< float > &src, const SimpleTensor< float > &biases, const TensorShape &dst_shape, const unsigned int num_groups) |
template SimpleTensor< half > | weights_reshape (const SimpleTensor< half > &src, const SimpleTensor< half > &biases, const TensorShape &dst_shape, const unsigned int num_groups) |
template SimpleTensor< uint8_t > | weights_reshape (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &biases, const TensorShape &dst_shape, const unsigned int num_groups) |
template<typename T > | |
SimpleTensor< T > | winograd_input_transform (const SimpleTensor< T > &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template<typename T > | |
SimpleTensor< T > | winograd_filter_transform (const SimpleTensor< T > &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template<typename T > | |
SimpleTensor< T > | winograd_output_transform (const SimpleTensor< T > &in, const SimpleTensor< T > &b, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template SimpleTensor< float > | winograd_filter_transform (const SimpleTensor< float > &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template SimpleTensor< float > | winograd_input_transform (const SimpleTensor< float > &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template SimpleTensor< float > | winograd_output_transform (const SimpleTensor< float > &in, const SimpleTensor< float > &b, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template SimpleTensor< half > | winograd_filter_transform (const SimpleTensor< half > &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template SimpleTensor< half > | winograd_input_transform (const SimpleTensor< half > &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template SimpleTensor< half > | winograd_output_transform (const SimpleTensor< half > &in, const SimpleTensor< half > &b, const TensorShape &output_shape, const WinogradInfo &winograd_info) |
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > | |
SimpleTensor< T > | yolo_layer (const SimpleTensor< T > &src, const ActivationLayerInfo &info, int32_t num_classes) |
template<> | |
SimpleTensor< uint8_t > | yolo_layer< uint8_t > (const SimpleTensor< uint8_t > &src, const ActivationLayerInfo &info, int32_t num_classes) |
template SimpleTensor< float > | yolo_layer (const SimpleTensor< float > &src, const ActivationLayerInfo &info, int32_t num_classes) |
template SimpleTensor< half > | yolo_layer (const SimpleTensor< half > &src, const ActivationLayerInfo &info, int32_t num_classes) |
|
strong |
|
strong |
|
strong |
Winograd transform type.
Enumerator | |
---|---|
INPUT | Winograd input transform. |
FILTER | Winograd filter transform. |
OUTPUT | Winograd output transform. |
Definition at line 40 of file Winograd.h.
SimpleTensor< T > absolute_difference | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2, | ||
DataType | dst_data_type | ||
) |
Definition at line 38 of file AbsoluteDifference.cpp.
References SimpleTensor< T >::num_elements(), arm_compute::test::saturate_cast(), and SimpleTensor< T >::shape().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::absolute_difference | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2, | ||
DataType | dst_data_type | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::absolute_difference | ( | const SimpleTensor< int16_t > & | src1, |
const SimpleTensor< int16_t > & | src2, | ||
DataType | dst_data_type | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::absolute_difference | ( | const SimpleTensor< int8_t > & | src1, |
const SimpleTensor< int8_t > & | src2, | ||
DataType | dst_data_type | ||
) |
SimpleTensor< T2 > accumulate | ( | const SimpleTensor< T1 > & | src, |
DataType | output_data_type | ||
) |
Definition at line 38 of file Accumulate.cpp.
References arm_compute::test::validation::dst, arm_compute::test::library, SimpleTensor< T >::num_elements(), arm_compute::test::saturate_cast(), and SimpleTensor< T >::shape().
Referenced by accumulate_squared().
template SimpleTensor<int16_t> arm_compute::test::validation::reference::accumulate | ( | const SimpleTensor< uint8_t > & | src, |
DataType | output_data_type | ||
) |
SimpleTensor< T2 > accumulate_squared | ( | const SimpleTensor< T1 > & | src, |
uint32_t | shift, | ||
DataType | output_data_type | ||
) |
Definition at line 80 of file Accumulate.cpp.
References accumulate(), accumulate_weighted(), ARM_COMPUTE_ERROR_ON_MSG, arm_compute::test::validation::dst, arm_compute::test::library, SimpleTensor< T >::num_elements(), arm_compute::test::saturate_cast(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<int16_t> arm_compute::test::validation::reference::accumulate_squared | ( | const SimpleTensor< uint8_t > & | src, |
uint32_t | shift, | ||
DataType | output_data_type | ||
) |
SimpleTensor< T2 > accumulate_weighted | ( | const SimpleTensor< T1 > & | src, |
float | alpha, | ||
DataType | output_data_type | ||
) |
Definition at line 58 of file Accumulate.cpp.
References ARM_COMPUTE_ERROR_ON_MSG, arm_compute::test::validation::dst, arm_compute::test::library, SimpleTensor< T >::num_elements(), and SimpleTensor< T >::shape().
Referenced by accumulate_squared().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::accumulate_weighted | ( | const SimpleTensor< uint8_t > & | src, |
float | alpha, | ||
DataType | output_data_type | ||
) |
|
inline |
Definition at line 39 of file ActivationLayer.h.
References ActivationLayerInfo::ABS, activation_layer(), ARM_COMPUTE_ERROR, arm_compute::test::validation::b, ActivationLayerInfo::BOUNDED_RELU, ActivationLayerInfo::ELU, ActivationLayerInfo::HARD_SWISH, ActivationLayerInfo::IDENTITY, arm_compute::test::validation::info, ActivationLayerInfo::LEAKY_RELU, ActivationLayerInfo::LINEAR, ActivationLayerInfo::LOGISTIC, ActivationLayerInfo::LU_BOUNDED_RELU, ActivationLayerInfo::RELU, ActivationLayerInfo::SOFT_RELU, ActivationLayerInfo::SQRT, ActivationLayerInfo::SQUARE, arm_compute::test::validation::src, and ActivationLayerInfo::TANH.
SimpleTensor< T > activation_layer | ( | const SimpleTensor< T > & | src, |
ActivationLayerInfo | info, | ||
const QuantizationInfo & | oq_info | ||
) |
Definition at line 38 of file ActivationLayer.cpp.
References ActivationLayerInfo::a(), ActivationLayerInfo::activation(), ARM_COMPUTE_UNUSED, arm_compute::test::validation::b, ActivationLayerInfo::b(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), and SimpleTensor< T >::shape().
Referenced by activate_float(), activation_layer< int16_t >(), and batch_normalization_layer().
template SimpleTensor<int32_t> arm_compute::test::validation::reference::activation_layer | ( | const SimpleTensor< int32_t > & | src, |
ActivationLayerInfo | info, | ||
const QuantizationInfo & | oq_info | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::activation_layer | ( | const SimpleTensor< float > & | src, |
ActivationLayerInfo | info, | ||
const QuantizationInfo & | oq_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::activation_layer | ( | const SimpleTensor< half > & | src, |
ActivationLayerInfo | info, | ||
const QuantizationInfo & | oq_info | ||
) |
SimpleTensor<int16_t> arm_compute::test::validation::reference::activation_layer< int16_t > | ( | const SimpleTensor< int16_t > & | src, |
ActivationLayerInfo | info, | ||
const QuantizationInfo & | oq_info | ||
) |
Definition at line 82 of file ActivationLayer.cpp.
References activation_layer(), arm_compute::test::validation::convert_from_symmetric(), arm_compute::test::validation::dst, QuantizationInfo::empty(), arm_compute::test::validation::info, and SimpleTensor< T >::quantization_info().
SimpleTensor<int8_t> arm_compute::test::validation::reference::activation_layer< int8_t > | ( | const SimpleTensor< int8_t > & | src, |
ActivationLayerInfo | info, | ||
const QuantizationInfo & | oq_info | ||
) |
Definition at line 71 of file ActivationLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::dst, QuantizationInfo::empty(), arm_compute::test::validation::info, and SimpleTensor< T >::quantization_info().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::activation_layer< uint8_t > | ( | const SimpleTensor< uint8_t > & | src, |
ActivationLayerInfo | info, | ||
const QuantizationInfo & | oq_info | ||
) |
Definition at line 60 of file ActivationLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::dst, QuantizationInfo::empty(), arm_compute::test::validation::info, and SimpleTensor< T >::quantization_info().
SimpleTensor< T > arithmetic_division | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2, | ||
DataType | data_type | ||
) |
Definition at line 79 of file ArithmeticDivision.cpp.
References TensorShape::broadcast_shape(), arm_compute::test::validation::data_type, arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
template SimpleTensor<half> arm_compute::test::validation::reference::arithmetic_division | ( | const SimpleTensor< half > & | src1, |
const SimpleTensor< half > & | src2, | ||
DataType | data_type | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::arithmetic_division | ( | const SimpleTensor< float > & | src1, |
const SimpleTensor< float > & | src2, | ||
DataType | data_type | ||
) |
SimpleTensor< T > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< T > & | src1, | ||
const SimpleTensor< T > & | src2, | ||
SimpleTensor< T > & | dst, | ||
ConvertPolicy | convert_policy | ||
) |
Definition at line 92 of file ArithmeticOperations.cpp.
References arm_compute::test::validation::dst.
Referenced by arithmetic_operation(), and laplacian_reconstruct().
SimpleTensor< uint8_t > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< uint8_t > & | src1, | ||
const SimpleTensor< uint8_t > & | src2, | ||
SimpleTensor< uint8_t > & | dst, | ||
ConvertPolicy | convert_policy | ||
) |
Definition at line 104 of file ArithmeticOperations.cpp.
References TensorShape::broadcast_shape(), arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::QASYMM8, SimpleTensor< T >::quantization_info(), and SimpleTensor< T >::shape().
SimpleTensor< int8_t > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< int8_t > & | src1, | ||
const SimpleTensor< int8_t > & | src2, | ||
SimpleTensor< int8_t > & | dst, | ||
ConvertPolicy | convert_policy | ||
) |
Definition at line 131 of file ArithmeticOperations.cpp.
References TensorShape::broadcast_shape(), arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::QASYMM8_SIGNED, SimpleTensor< T >::quantization_info(), and SimpleTensor< T >::shape().
SimpleTensor< int16_t > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< int16_t > & | src1, | ||
const SimpleTensor< int16_t > & | src2, | ||
SimpleTensor< int16_t > & | dst, | ||
ConvertPolicy | convert_policy | ||
) |
Definition at line 158 of file ArithmeticOperations.cpp.
References arithmetic_operation(), TensorShape::broadcast_shape(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::QSYMM16, SimpleTensor< T >::quantization_info(), and SimpleTensor< T >::shape().
template SimpleTensor< int32_t > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< int32_t > & | src1, | ||
const SimpleTensor< int32_t > & | src2, | ||
SimpleTensor< int32_t > & | dst, | ||
ConvertPolicy | convert_policy | ||
) |
template SimpleTensor< half > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< half > & | src1, | ||
const SimpleTensor< half > & | src2, | ||
SimpleTensor< half > & | dst, | ||
ConvertPolicy | convert_policy | ||
) |
template SimpleTensor< float > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< float > & | src1, | ||
const SimpleTensor< float > & | src2, | ||
SimpleTensor< float > & | dst, | ||
ConvertPolicy | convert_policy | ||
) |
SimpleTensor< T > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< T > & | src1, | ||
const SimpleTensor< T > & | src2, | ||
DataType | dst_data_type, | ||
ConvertPolicy | convert_policy | ||
) |
Definition at line 188 of file ArithmeticOperations.cpp.
References arithmetic_operation(), ARM_COMPUTE_ERROR_ON_MSG, TensorShape::broadcast_shape(), arm_compute::test::validation::dst, arm_compute::is_data_type_quantized(), and SimpleTensor< T >::shape().
template SimpleTensor< int32_t > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< int32_t > & | src1, | ||
const SimpleTensor< int32_t > & | src2, | ||
DataType | dst_data_type, | ||
ConvertPolicy | convert_policy | ||
) |
template SimpleTensor< int16_t > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< int16_t > & | src1, | ||
const SimpleTensor< int16_t > & | src2, | ||
DataType | dst_data_type, | ||
ConvertPolicy | convert_policy | ||
) |
template SimpleTensor< int8_t > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< int8_t > & | src1, | ||
const SimpleTensor< int8_t > & | src2, | ||
DataType | dst_data_type, | ||
ConvertPolicy | convert_policy | ||
) |
template SimpleTensor< half > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< half > & | src1, | ||
const SimpleTensor< half > & | src2, | ||
DataType | dst_data_type, | ||
ConvertPolicy | convert_policy | ||
) |
template SimpleTensor< float > arithmetic_operation | ( | ArithmeticOperation | op, |
const SimpleTensor< float > & | src1, | ||
const SimpleTensor< float > & | src2, | ||
DataType | dst_data_type, | ||
ConvertPolicy | convert_policy | ||
) |
SimpleTensor< T > batch_normalization_layer | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< T > & | mean, | ||
const SimpleTensor< T > & | var, | ||
const SimpleTensor< T > & | beta, | ||
const SimpleTensor< T > & | gamma, | ||
float | epsilon, | ||
ActivationLayerInfo | act_info | ||
) |
Definition at line 40 of file BatchNormalizationLayer.cpp.
References activation_layer(), caffe_mnist_image_extractor::cols, SimpleTensor< T >::data_type(), ActivationLayerInfo::enabled(), arm_compute::quantization::epsilon, caffe_mnist_image_extractor::rows, SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::batch_normalization_layer | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | mean, | ||
const SimpleTensor< float > & | var, | ||
const SimpleTensor< float > & | beta, | ||
const SimpleTensor< float > & | gamma, | ||
float | epsilon, | ||
ActivationLayerInfo | act_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::batch_normalization_layer | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | mean, | ||
const SimpleTensor< half > & | var, | ||
const SimpleTensor< half > & | beta, | ||
const SimpleTensor< half > & | gamma, | ||
float | epsilon, | ||
ActivationLayerInfo | act_info | ||
) |
SimpleTensor< T > batch_to_space | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< int32_t > & | block_shape, | ||
const TensorShape & | dst_shape | ||
) |
Definition at line 38 of file BatchToSpaceLayer.cpp.
References ARM_COMPUTE_ERROR_ON, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst_shape, SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::batch_to_space | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< int32_t > & | block_shape, | ||
const TensorShape & | dst_shape | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::batch_to_space | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< int32_t > & | block_shape, | ||
const TensorShape & | dst_shape | ||
) |
SimpleTensor< T > bitwise_and | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2 | ||
) |
Definition at line 35 of file BitwiseAnd.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), and SimpleTensor< T >::shape().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::bitwise_and | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2 | ||
) |
SimpleTensor< T > bitwise_not | ( | const SimpleTensor< T > & | src | ) |
Definition at line 35 of file BitwiseNot.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::bitwise_not | ( | const SimpleTensor< uint8_t > & | src | ) |
SimpleTensor< T > bitwise_or | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2 | ||
) |
Definition at line 35 of file BitwiseOr.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), and SimpleTensor< T >::shape().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::bitwise_or | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2 | ||
) |
SimpleTensor< T > bitwise_xor | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2 | ||
) |
Definition at line 35 of file BitwiseXor.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), and SimpleTensor< T >::shape().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::bitwise_xor | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2 | ||
) |
SimpleTensor< T > bounding_box_transform | ( | const SimpleTensor< T > & | boxes, |
const SimpleTensor< TDeltas > & | deltas, | ||
const BoundingBoxTransformInfo & | info | ||
) |
Definition at line 40 of file BoundingBoxTransform.cpp.
References BoundingBoxTransformInfo::apply_scale(), ARM_COMPUTE_ERROR_ON, BoundingBoxTransformInfo::bbox_xform_clip(), BoundingBoxTransformInfo::correct_transform_coords(), SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), BoundingBoxTransformInfo::img_height(), BoundingBoxTransformInfo::img_width(), arm_compute::test::validation::info, offset(), BoundingBoxTransformInfo::scale(), SimpleTensor< T >::shape(), and BoundingBoxTransformInfo::weights().
template SimpleTensor<float> arm_compute::test::validation::reference::bounding_box_transform | ( | const SimpleTensor< float > & | boxes, |
const SimpleTensor< float > & | deltas, | ||
const BoundingBoxTransformInfo & | info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::bounding_box_transform | ( | const SimpleTensor< half > & | boxes, |
const SimpleTensor< half > & | deltas, | ||
const BoundingBoxTransformInfo & | info | ||
) |
SimpleTensor<uint16_t> arm_compute::test::validation::reference::bounding_box_transform | ( | const SimpleTensor< uint16_t > & | boxes, |
const SimpleTensor< uint8_t > & | deltas, | ||
const BoundingBoxTransformInfo & | info | ||
) |
Definition at line 105 of file BoundingBoxTransform.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::info, and SimpleTensor< T >::quantization_info().
SimpleTensor< T > box3x3 | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 38 of file Box3x3.cpp.
References arm_compute::test::validation::apply_2d_spatial_filter(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), scale(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and arm_compute::utils::cast::U.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::box3x3 | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > canny_edge_detector | ( | const SimpleTensor< T > & | src, |
int32_t | upper_thresh, | ||
int32_t | lower_thresh, | ||
int | gradient_size, | ||
MagnitudeType | norm_type, | ||
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 234 of file CannyEdgeDetector.cpp.
References arm_compute::test::validation::src.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::canny_edge_detector | ( | const SimpleTensor< uint8_t > & | src, |
int32_t | upper_thresh, | ||
int32_t | lower_thresh, | ||
int | gradient_size, | ||
MagnitudeType | norm_type, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
std::vector< SimpleTensor< T > > channel_combine | ( | const TensorShape & | shape, |
const std::vector< SimpleTensor< T >> & | image_planes, | ||
Format | format | ||
) |
Definition at line 91 of file ChannelCombine.cpp.
References ARM_COMPUTE_ERROR, arm_compute::test::validation::dst, arm_compute::test::index2coord(), arm_compute::IYUV, SimpleTensor< T >::num_channels(), SimpleTensor< T >::num_elements(), arm_compute::NV12, arm_compute::NV21, arm_compute::RGB888, arm_compute::RGBA8888, arm_compute::test::validation::shape, SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::utils::cast::U, arm_compute::UYVY422, Dimensions< T >::x(), Dimensions< T >::y(), arm_compute::YUV444, and arm_compute::YUYV422.
template std::vector<SimpleTensor<uint8_t> > arm_compute::test::validation::reference::channel_combine | ( | const TensorShape & | shape, |
const std::vector< SimpleTensor< uint8_t >> & | image_planes, | ||
Format | format | ||
) |
SimpleTensor< uint8_t > channel_extract | ( | const TensorShape & | shape, |
const std::vector< SimpleTensor< T >> & | tensor_planes, | ||
Format | format, | ||
Channel | channel | ||
) |
Definition at line 38 of file ChannelExtract.cpp.
References arm_compute::calculate_subsampled_shape(), arm_compute::channel_idx_from_format(), arm_compute::test::validation::dst, arm_compute::plane_idx_from_channel(), arm_compute::test::validation::shape, arm_compute::test::validation::src, arm_compute::U8, arm_compute::UYVY422, arm_compute::Y, and arm_compute::YUYV422.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::channel_extract | ( | const TensorShape & | shape, |
const std::vector< SimpleTensor< uint8_t >> & | tensor_planes, | ||
Format | format, | ||
Channel | channel | ||
) |
SimpleTensor< T > channel_shuffle | ( | const SimpleTensor< T > & | src, |
int | num_groups | ||
) |
Definition at line 39 of file ChannelShuffle.cpp.
References batches, copy(), SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_ref, M, N, SimpleTensor< T >::num_channels(), arm_compute::test::validation::num_groups, SimpleTensor< T >::quantization_info(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and arm_compute::test::validation::src_ref.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::channel_shuffle | ( | const SimpleTensor< uint8_t > & | src, |
int | num_groups | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::channel_shuffle | ( | const SimpleTensor< uint16_t > & | src, |
int | num_groups | ||
) |
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::channel_shuffle | ( | const SimpleTensor< uint32_t > & | src, |
int | num_groups | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::channel_shuffle | ( | const SimpleTensor< half > & | src, |
int | num_groups | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::channel_shuffle | ( | const SimpleTensor< float > & | src, |
int | num_groups | ||
) |
SimpleTensor< T > col2im | ( | const SimpleTensor< T > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | num_groups | ||
) |
Definition at line 38 of file Col2Im.cpp.
References arm_compute::test::validation::b, batches, arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, arm_compute::test::validation::num_groups, SimpleTensor< T >::shape(), arm_compute::test::validation::src, TensorShape::total_size(), Dimensions< T >::x(), Dimensions< T >::y(), and Dimensions< T >::z().
template SimpleTensor<float> arm_compute::test::validation::reference::col2im | ( | const SimpleTensor< float > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | num_groups | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::col2im | ( | const SimpleTensor< half > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | num_groups | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::col2im | ( | const SimpleTensor< uint8_t > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | num_groups | ||
) |
std::vector< SimpleTensor< T > > color_convert | ( | const TensorShape & | shape, |
const std::vector< SimpleTensor< T >> & | tensor_planes, | ||
Format | src_format, | ||
Format | dst_format | ||
) |
Definition at line 93 of file ColorConvert.cpp.
References ARM_COMPUTE_ERROR, arm_compute::test::colorconvert_helper::detail::colorconvert_iyuv_to_rgb(), arm_compute::test::colorconvert_helper::detail::colorconvert_nv12_to_rgb(), arm_compute::test::colorconvert_helper::detail::colorconvert_nv_to_iyuv(), arm_compute::test::colorconvert_helper::detail::colorconvert_nv_to_yuv4(), arm_compute::test::colorconvert_helper::detail::colorconvert_rgb_to_iyuv(), arm_compute::test::colorconvert_helper::detail::colorconvert_rgb_to_nv12(), arm_compute::test::colorconvert_helper::detail::colorconvert_rgb_to_rgbx(), arm_compute::test::colorconvert_helper::detail::colorconvert_rgb_to_u8(), arm_compute::test::colorconvert_helper::detail::colorconvert_rgb_to_yuv4(), arm_compute::test::colorconvert_helper::detail::colorconvert_rgbx_to_rgb(), arm_compute::test::colorconvert_helper::detail::colorconvert_yuyv_to_iyuv(), arm_compute::test::colorconvert_helper::detail::colorconvert_yuyv_to_nv12(), arm_compute::test::colorconvert_helper::detail::colorconvert_yuyv_to_rgb(), arm_compute::test::validation::dst, arm_compute::IYUV, arm_compute::NV12, arm_compute::NV21, arm_compute::RGB888, arm_compute::RGBA8888, arm_compute::test::validation::shape, arm_compute::U8, arm_compute::UYVY422, arm_compute::YUV444, and arm_compute::YUYV422.
template std::vector<SimpleTensor<uint8_t> > arm_compute::test::validation::reference::color_convert | ( | const TensorShape & | shape, |
const std::vector< SimpleTensor< uint8_t >> & | tensor_planes, | ||
Format | src_format, | ||
Format | dst_format | ||
) |
SimpleTensor< uint8_t > compare | ( | ComparisonOperation | op, |
const SimpleTensor< T > & | src1, | ||
const SimpleTensor< T > & | src2 | ||
) |
Definition at line 111 of file Comparisons.cpp.
References TensorShape::broadcast_shape(), arm_compute::test::validation::dst, SimpleTensor< T >::shape(), and arm_compute::U8.
Referenced by compare().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::compare | ( | ComparisonOperation | op, |
const SimpleTensor< uint8_t > & | src1, | ||
const SimpleTensor< uint8_t > & | src2 | ||
) |
Definition at line 123 of file Comparisons.cpp.
References TensorShape::broadcast_shape(), arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::QASYMM8, SimpleTensor< T >::shape(), and arm_compute::U8.
SimpleTensor<uint8_t> arm_compute::test::validation::reference::compare | ( | ComparisonOperation | op, |
const SimpleTensor< int8_t > & | src1, | ||
const SimpleTensor< int8_t > & | src2 | ||
) |
Definition at line 146 of file Comparisons.cpp.
References TensorShape::broadcast_shape(), compare(), arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::QASYMM8_SIGNED, SimpleTensor< T >::shape(), and arm_compute::U8.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::compare | ( | ComparisonOperation | op, |
const SimpleTensor< half > & | src1, | ||
const SimpleTensor< half > & | src2 | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::compare | ( | ComparisonOperation | op, |
const SimpleTensor< float > & | src1, | ||
const SimpleTensor< float > & | src2 | ||
) |
SimpleTensor< T > compute_all_anchors | ( | const SimpleTensor< T > & | anchors, |
const ComputeAnchorsInfo & | info | ||
) |
Definition at line 40 of file ComputeAllAnchors.cpp.
References SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), ComputeAnchorsInfo::feat_height(), ComputeAnchorsInfo::feat_width(), arm_compute::test::validation::info, SimpleTensor< T >::shape(), and ComputeAnchorsInfo::spatial_scale().
Referenced by compute_all_anchors().
template SimpleTensor<float> arm_compute::test::validation::reference::compute_all_anchors | ( | const SimpleTensor< float > & | anchors, |
const ComputeAnchorsInfo & | info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::compute_all_anchors | ( | const SimpleTensor< half > & | anchors, |
const ComputeAnchorsInfo & | info | ||
) |
SimpleTensor<int16_t> arm_compute::test::validation::reference::compute_all_anchors | ( | const SimpleTensor< int16_t > & | anchors, |
const ComputeAnchorsInfo & | info | ||
) |
Definition at line 81 of file ComputeAllAnchors.cpp.
References compute_all_anchors(), arm_compute::test::validation::convert_from_symmetric(), and SimpleTensor< T >::quantization_info().
void arm_compute::test::validation::reference::compute_min_max | ( | const SimpleTensor< T > & | src, |
T & | min, | ||
T & | max | ||
) |
Definition at line 35 of file MinMaxLocation.cpp.
References ARM_COMPUTE_ERROR_ON, and SimpleTensor< T >::num_elements().
SimpleTensor<OT> arm_compute::test::validation::reference::compute_reduction_operation | ( | const SimpleTensor< T > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | axis, | ||
ReductionOperation | op | ||
) |
Definition at line 178 of file ReductionOperation.cpp.
References arm_compute::ARG_IDX_MAX, arm_compute::ARG_IDX_MIN, ARM_COMPUTE_ERROR, SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, SimpleTensor< T >::quantization_info(), arm_compute::S32, and SimpleTensor< T >::shape().
SimpleTensor< T > concatenate_layer | ( | std::vector< SimpleTensor< T >> & | srcs, |
SimpleTensor< T > & | dst, | ||
unsigned int | axis | ||
) |
Definition at line 116 of file ConcatenateLayer.cpp.
References ARM_COMPUTE_ERROR, Window::DimX, Window::DimY, Window::DimZ, arm_compute::test::validation::dst, tf_frozen_model_extractor::t, and arm_compute::utils::cast::U.
template SimpleTensor<float> arm_compute::test::validation::reference::concatenate_layer | ( | std::vector< SimpleTensor< float >> & | srcs, |
SimpleTensor< float > & | dst, | ||
unsigned int | axis | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::concatenate_layer | ( | std::vector< SimpleTensor< half >> & | srcs, |
SimpleTensor< half > & | dst, | ||
unsigned int | axis | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::concatenate_layer | ( | std::vector< SimpleTensor< uint8_t >> & | srcs, |
SimpleTensor< uint8_t > & | dst, | ||
unsigned int | axis | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::concatenate_layer | ( | std::vector< SimpleTensor< int8_t >> & | srcs, |
SimpleTensor< int8_t > & | dst, | ||
unsigned int | axis | ||
) |
SimpleTensor< T > conv2d_dft | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< T > & | w, | ||
const PadStrideInfo & | conv_info | ||
) |
Performs and DFT based convolution on a real input.
[in] | src | Source tensor. |
[in] | w | Weights tensor. |
[in] | conv_info | Convolution related metadata. |
Definition at line 396 of file DFT.cpp.
References arm_compute::test::validation::conv_info, copy(), dft_1d(), dft_2d(), PadStrideInfo::pad_bottom(), pad_layer(), PadStrideInfo::pad_left(), PadStrideInfo::pad_right(), PadStrideInfo::pad_top(), rdft_1d(), rdft_2d(), reverse(), ridft_1d(), ridft_2d(), SimpleTensor< T >::shape(), slice(), arm_compute::utils::cast::U, arm_compute::U32, and arm_compute::test::validation::w.
template SimpleTensor<float> arm_compute::test::validation::reference::conv2d_dft | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | w, | ||
const PadStrideInfo & | conv_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::conv2d_dft | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | w, | ||
const PadStrideInfo & | conv_info | ||
) |
SimpleTensor< T > convert_fully_connected_weights | ( | const SimpleTensor< T > & | src, |
const TensorShape & | original_input_shape, | ||
const DataLayout | training_data_layout | ||
) |
Definition at line 35 of file ConvertFullyConnectedWeights.cpp.
References arm_compute::CHANNEL, arm_compute::coords2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::get_data_layout_dimension_index(), arm_compute::HEIGHT, arm_compute::index2coords(), arm_compute::NCHW, arm_compute::NHWC, SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::WIDTH, Dimensions< T >::x(), and Dimensions< T >::y().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::convert_fully_connected_weights | ( | const SimpleTensor< uint8_t > & | src, |
const TensorShape & | original_input_shape, | ||
const DataLayout | training_data_layout | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::convert_fully_connected_weights | ( | const SimpleTensor< half > & | src, |
const TensorShape & | original_input_shape, | ||
const DataLayout | training_data_layout | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::convert_fully_connected_weights | ( | const SimpleTensor< float > & | src, |
const TensorShape & | original_input_shape, | ||
const DataLayout | training_data_layout | ||
) |
SimpleTensor< T > convolution | ( | const SimpleTensor< uint8_t > & | src, |
DataType | output_data_type, | ||
const int16_t * | conv, | ||
uint32_t | scale, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
const unsigned int | width, | ||
const unsigned int | height | ||
) |
Definition at line 38 of file Convolution.cpp.
References arm_compute::test::validation::apply_2d_spatial_filter(), ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), arm_compute::test::saturate_cast(), scale(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and sum().
SimpleTensor< T > convolution_layer | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< TW > & | weights, | ||
const SimpleTensor< TB > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
const Size2D & | dilation, | ||
unsigned int | num_groups, | ||
QuantizationInfo | out_quant_info | ||
) |
Definition at line 111 of file ConvolutionLayer.cpp.
References convolution_layer_nchw(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::info, arm_compute::test::validation::num_groups, arm_compute::test::validation::output_shape, SimpleTensor< T >::quantization_info(), and arm_compute::test::validation::src.
Referenced by deconvolution_layer().
template SimpleTensor<float> arm_compute::test::validation::reference::convolution_layer | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | weights, | ||
const SimpleTensor< float > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
const Size2D & | dilation, | ||
unsigned int | num_groups, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::convolution_layer | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | weights, | ||
const SimpleTensor< half > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
const Size2D & | dilation, | ||
unsigned int | num_groups, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::convolution_layer | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint8_t > & | weights, | ||
const SimpleTensor< int32_t > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
const Size2D & | dilation, | ||
unsigned int | num_groups, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::convolution_layer | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< int8_t > & | weights, | ||
const SimpleTensor< int32_t > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
const Size2D & | dilation, | ||
unsigned int | num_groups, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::convolution_layer | ( | const SimpleTensor< int8_t > & | src, |
const SimpleTensor< int8_t > & | weights, | ||
const SimpleTensor< int32_t > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
const Size2D & | dilation, | ||
unsigned int | num_groups, | ||
QuantizationInfo | out_quant_info | ||
) |
SimpleTensor<T> arm_compute::test::validation::reference::convolution_layer_nchw | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< TW > & | weights, | ||
const SimpleTensor< TB > & | bias, | ||
SimpleTensor< T > & | dst, | ||
const PadStrideInfo & | info, | ||
const Size2D & | dilation, | ||
unsigned int | num_groups | ||
) |
Definition at line 45 of file ConvolutionLayer.cpp.
References ARM_COMPUTE_ASSERT, ARM_COMPUTE_ERROR_ON, arm_compute::test::convolution_3d::detail::convolution3d(), arm_compute::test::validation::dst, arm_compute::test::validation::num_groups, arm_compute::test::validation::output_wh, PadStrideInfo::pad_left(), PadStrideInfo::pad_top(), arm_compute::scaled_dimensions(), SimpleTensor< T >::shape(), PadStrideInfo::stride(), Size2D::x(), and Size2D::y().
Referenced by convolution_layer().
SimpleTensor< T > copy | ( | const SimpleTensor< T > & | src, |
const TensorShape & | output_shape | ||
) |
Definition at line 37 of file Copy.cpp.
References ARM_COMPUTE_ERROR_ON_MISMATCHING_DIMENSIONS, SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), arm_compute::test::validation::output_shape, SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
Referenced by channel_shuffle(), Dimensions< size_t >::collapse(), TensorShape::collapsed_from(), conv2d_dft(), depthconcatenate_layer(), flatten_layer(), gemm_reshape_lhs_matrix(), gemm_reshape_rhs_matrix(), arm_compute::test::validation::get_tile(), Dimensions< size_t >::remove(), TensorShape::remove_dimension(), and TopNPredictionsAccessor::TopNPredictionsAccessor().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< uint8_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< int8_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< uint16_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< int16_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< uint32_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< int32_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< half > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::copy | ( | const SimpleTensor< float > & | src, |
const TensorShape & | output_shape | ||
) |
SimpleTensor< float > crop_and_resize | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
Definition at line 160 of file CropResize.cpp.
References ARM_COMPUTE_ERROR_ON, SimpleTensor< T >::data(), SimpleTensor< T >::data_layout(), arm_compute::F32, arm_compute::NHWC, SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), Coordinates2D::x, and Coordinates2D::y.
template SimpleTensor<float> arm_compute::test::validation::reference::crop_and_resize | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::crop_and_resize | ( | const SimpleTensor< uint16_t > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::crop_and_resize | ( | const SimpleTensor< uint32_t > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::crop_and_resize | ( | const SimpleTensor< int16_t > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::crop_and_resize | ( | const SimpleTensor< int32_t > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::crop_and_resize | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::crop_and_resize | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< float > & | boxes, | ||
SimpleTensor< int32_t > | box_ind, | ||
Coordinates2D | crop_size, | ||
InterpolationPolicy | method, | ||
float | extrapolation_value | ||
) |
SimpleTensor< T > deconvolution_layer | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< T > & | weights, | ||
const SimpleTensor< TB > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
QuantizationInfo | out_qinfo = QuantizationInfo() |
||
) |
Deconvolution reference implementation.
src Input tensor. 3 lower dimensions represent a single input, and an optional 4th dimension for batch of inputs. Data types supported: QASYMM8/QASYMM8_SIGNED/F32/F16. weights The 4d weights with dimensions [width, height, OFM, IFM]. Data type supported: Same as input
. bias Optional, ignored if NULL. The biases have one dimension. Data type supported: Same as input
, except for input of QASYMM8 and QASYMM8_SIGNED type where biases should be of S32 type output_shape Output tensor shape. The output has the same number of dimensions as the input
. info Contains padding and policies to be used in the deconvolution, this is decribed in PadStrideInfo. a The number of zeros added to right and top edges of the input.
Definition at line 37 of file DeconvolutionLayer.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::CEIL, arm_compute::test::validation::conv_info, convolution_layer(), SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::info, offset(), arm_compute::test::validation::output_shape, PadStrideInfo::pad_bottom(), PadStrideInfo::pad_left(), PadStrideInfo::pad_right(), PadStrideInfo::pad_top(), arm_compute::QASYMM8, arm_compute::QASYMM8_SIGNED, SimpleTensor< T >::quantization_info(), TensorShape::set(), SimpleTensor< T >::shape(), slice(), arm_compute::test::validation::src, PadStrideInfo::stride(), arm_compute::utils::cast::U, weights_height, weights_width, Dimensions< T >::x(), and Dimensions< T >::y().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::deconvolution_layer | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint8_t > & | weights, | ||
const SimpleTensor< int32_t > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::deconvolution_layer | ( | const SimpleTensor< int8_t > & | src, |
const SimpleTensor< int8_t > & | weights, | ||
const SimpleTensor< int32_t > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::deconvolution_layer | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | weights, | ||
const SimpleTensor< float > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::deconvolution_layer | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | weights, | ||
const SimpleTensor< half > & | bias, | ||
const TensorShape & | output_shape, | ||
const PadStrideInfo & | info, | ||
QuantizationInfo | out_quant_info | ||
) |
SimpleTensor< T2 > depth_convert | ( | const SimpleTensor< T1 > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
Definition at line 42 of file DepthConvertLayer.cpp.
References SimpleTensor< T >::data_type(), arm_compute::element_size_from_data_type(), SimpleTensor< T >::num_elements(), arm_compute::SATURATE, and SimpleTensor< T >::shape().
template SimpleTensor< float > depth_convert | ( | const SimpleTensor< uint8_t > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor< float > depth_convert | ( | const SimpleTensor< int8_t > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor< float > depth_convert | ( | const SimpleTensor< uint16_t > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor< float > depth_convert | ( | const SimpleTensor< int16_t > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor< float > depth_convert | ( | const SimpleTensor< uint32_t > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor< float > depth_convert | ( | const SimpleTensor< int32_t > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::depth_convert | ( | const SimpleTensor< bfloat16 > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor< float > depth_convert | ( | const SimpleTensor< half > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
template SimpleTensor< bfloat16 > depth_convert | ( | const SimpleTensor< float > & | src, |
DataType | dt_out, | ||
ConvertPolicy | policy, | ||
uint32_t | shift | ||
) |
SimpleTensor< T > depth_to_space | ( | const SimpleTensor< T > & | src, |
const TensorShape & | dst_shape, | ||
int32_t | block_shape | ||
) |
Definition at line 38 of file DepthToSpaceLayer.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::b, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst_shape, SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::depth_to_space | ( | const SimpleTensor< float > & | src, |
const TensorShape & | dst_shape, | ||
int32_t | block_shape | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::depth_to_space | ( | const SimpleTensor< half > & | src, |
const TensorShape & | dst_shape, | ||
int32_t | block_shape | ||
) |
SimpleTensor< T > depthconcatenate_layer | ( | const std::vector< SimpleTensor< T >> & | srcs, |
SimpleTensor< T > & | dst | ||
) |
Definition at line 37 of file DepthConcatenateLayer.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::b, batches, copy(), SimpleTensor< T >::data(), arm_compute::test::validation::data_type, arm_compute::dequantize_qasymm8(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), arm_compute::QASYMM8, SimpleTensor< T >::quantization_info(), arm_compute::quantize_qasymm8(), SimpleTensor< T >::shape(), slice(), arm_compute::test::validation::src, and tf_frozen_model_extractor::t.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::depthconcatenate_layer | ( | const std::vector< SimpleTensor< uint8_t >> & | srcs, |
SimpleTensor< uint8_t > & | dst | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::depthconcatenate_layer | ( | const std::vector< SimpleTensor< float >> & | srcs, |
SimpleTensor< float > & | dst | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::depthconcatenate_layer | ( | const std::vector< SimpleTensor< half >> & | srcs, |
SimpleTensor< half > & | dst | ||
) |
SimpleTensor<T> arm_compute::test::validation::reference::depthwise_convolution | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< TW > & | weights, | ||
const SimpleTensor< TB > & | biases, | ||
const TensorShape & | dst_shape, | ||
const PadStrideInfo & | conv_info, | ||
unsigned int | depth_multiplier, | ||
const Size2D & | dilation = Size2D(1U, 1U) , |
||
const QuantizationInfo & | out_quant_info = QuantizationInfo(0.0f, 0) |
||
) |
SimpleTensor<float> arm_compute::test::validation::reference::depthwise_convolution | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | weights, | ||
const SimpleTensor< float > & | biases, | ||
const TensorShape & | dst_shape, | ||
const PadStrideInfo & | conv_info, | ||
unsigned int | depth_multiplier, | ||
const Size2D & | dilation, | ||
const QuantizationInfo & | out_quant_info | ||
) |
Definition at line 238 of file DepthwiseConvolutionLayer.cpp.
SimpleTensor<half> arm_compute::test::validation::reference::depthwise_convolution | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | weights, | ||
const SimpleTensor< half > & | biases, | ||
const TensorShape & | dst_shape, | ||
const PadStrideInfo & | conv_info, | ||
unsigned int | depth_multiplier, | ||
const Size2D & | dilation, | ||
const QuantizationInfo & | out_quant_info | ||
) |
Definition at line 245 of file DepthwiseConvolutionLayer.cpp.
SimpleTensor<uint8_t> arm_compute::test::validation::reference::depthwise_convolution | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint8_t > & | weights, | ||
const SimpleTensor< int32_t > & | biases, | ||
const TensorShape & | dst_shape, | ||
const PadStrideInfo & | conv_info, | ||
unsigned int | depth_multiplier, | ||
const Size2D & | dilation, | ||
const QuantizationInfo & | out_quant_info | ||
) |
Definition at line 252 of file DepthwiseConvolutionLayer.cpp.
References arm_compute::test::validation::conv_info, arm_compute::test::validation::dst_shape, and arm_compute::test::validation::src.
SimpleTensor<uint8_t> arm_compute::test::validation::reference::depthwise_convolution | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< int8_t > & | weights, | ||
const SimpleTensor< int32_t > & | biases, | ||
const TensorShape & | dst_shape, | ||
const PadStrideInfo & | conv_info, | ||
unsigned int | depth_multiplier, | ||
const Size2D & | dilation, | ||
const QuantizationInfo & | out_quant_info | ||
) |
Definition at line 259 of file DepthwiseConvolutionLayer.cpp.
References arm_compute::test::validation::conv_info, arm_compute::test::validation::dst_shape, and arm_compute::test::validation::src.
SimpleTensor<int8_t> arm_compute::test::validation::reference::depthwise_convolution | ( | const SimpleTensor< int8_t > & | src, |
const SimpleTensor< int8_t > & | weights, | ||
const SimpleTensor< int32_t > & | biases, | ||
const TensorShape & | dst_shape, | ||
const PadStrideInfo & | conv_info, | ||
unsigned int | depth_multiplier, | ||
const Size2D & | dilation, | ||
const QuantizationInfo & | out_quant_info | ||
) |
Definition at line 266 of file DepthwiseConvolutionLayer.cpp.
References arm_compute::test::validation::conv_info, arm_compute::test::validation::dst_shape, and arm_compute::test::validation::src.
SimpleTensor<T> arm_compute::test::validation::reference::depthwise_separable_convolution_layer | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< T > & | depthwise_weights, | ||
const SimpleTensor< T > & | depthwise_biases, | ||
const TensorShape & | depthwise_out_shape, | ||
const SimpleTensor< T > & | pointwise_weights, | ||
const SimpleTensor< T > & | pointwise_biases, | ||
const TensorShape & | dst_shape, | ||
const PadStrideInfo & | depthwise_conv_info, | ||
const PadStrideInfo & | pointwise_conv_info | ||
) |
SimpleTensor< TOut > dequantization_layer | ( | const SimpleTensor< TIn > & | src | ) |
Definition at line 64 of file DequantizationLayer.cpp.
References SimpleTensor< T >::data_type().
template SimpleTensor< float > dequantization_layer | ( | const SimpleTensor< uint8_t > & | src | ) |
template SimpleTensor< float > dequantization_layer | ( | const SimpleTensor< int8_t > & | src | ) |
template SimpleTensor< float > dequantization_layer | ( | const SimpleTensor< int16_t > & | src | ) |
std::pair< SimpleTensor< T >, SimpleTensor< T > > derivative | ( | const SimpleTensor< U > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
GradientDimension | gradient_dimension | ||
) |
Definition at line 55 of file Derivative.cpp.
References SimpleTensor< T >::shape().
template std::pair<SimpleTensor<int16_t>, SimpleTensor<int16_t> > arm_compute::test::validation::reference::derivative | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
GradientDimension | gradient_dimension | ||
) |
void arm_compute::test::validation::reference::detection_windows_non_maxima_suppression | ( | std::vector< DetectionWindow > & | multi_windows, |
float | min_distance | ||
) |
Definition at line 60 of file HOGMultiDetection.cpp.
References DetectionWindow::height, DetectionWindow::idx_class, DetectionWindow::score, DetectionWindow::width, DetectionWindow::x, and DetectionWindow::y.
Referenced by hog_multi_detection().
SimpleTensor< T > dft_1d | ( | const SimpleTensor< T > & | src, |
FFTDirection | direction | ||
) |
Performs an one dimensional DFT on a complex input.
[in] | src | Source tensor. |
[in] | direction | Direction of the DFT. |
Definition at line 330 of file DFT.cpp.
References arm_compute::test::validation::dst, Inverse, and scale().
Referenced by conv2d_dft(), and arm_compute::test::validation::DATA_TEST_CASE().
template SimpleTensor<float> arm_compute::test::validation::reference::dft_1d | ( | const SimpleTensor< float > & | src, |
FFTDirection | direction | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::dft_1d | ( | const SimpleTensor< half > & | src, |
FFTDirection | direction | ||
) |
SimpleTensor< T > dft_2d | ( | const SimpleTensor< T > & | src, |
FFTDirection | direction | ||
) |
Performs a two dimensional DFT on a complex input.
[in] | src | Source tensor. |
[in] | direction | Direction of the DFT. |
Definition at line 370 of file DFT.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::dst, Forward, SimpleTensor< T >::num_channels(), permute(), scale(), and arm_compute::utils::cast::U.
Referenced by conv2d_dft(), and arm_compute::test::validation::DATA_TEST_CASE().
template SimpleTensor<float> arm_compute::test::validation::reference::dft_2d | ( | const SimpleTensor< float > & | src, |
FFTDirection | direction | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::dft_2d | ( | const SimpleTensor< half > & | src, |
FFTDirection | direction | ||
) |
SimpleTensor< T > dilate | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 41 of file Dilate.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), Dimensions< T >::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), Dimensions< T >::x(), and Dimensions< T >::y().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::dilate | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > elementwise_unary | ( | const SimpleTensor< T > & | src, |
ElementWiseUnary | op | ||
) |
Definition at line 35 of file ElementWiseUnary.cpp.
References arm_compute::ABS, ARM_COMPUTE_ERROR, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::EXP, arm_compute::LOG, arm_compute::support::cpp11::nearbyint(), arm_compute::NEG, SimpleTensor< T >::num_elements(), arm_compute::ROUND, arm_compute::RSQRT, SimpleTensor< T >::shape(), arm_compute::SIN, and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::elementwise_unary | ( | const SimpleTensor< float > & | src, |
ElementWiseUnary | op | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::elementwise_unary | ( | const SimpleTensor< half > & | src, |
ElementWiseUnary | op | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::elementwise_unary | ( | const SimpleTensor< int32_t > & | src, |
ElementWiseUnary | op | ||
) |
SimpleTensor< T > equalize_histogram | ( | const SimpleTensor< T > & | src | ) |
Definition at line 35 of file EqualizeHistogram.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::equalize_histogram | ( | const SimpleTensor< uint8_t > & | src | ) |
SimpleTensor< T > erode | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 41 of file Erode.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), Dimensions< T >::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), Dimensions< T >::x(), and Dimensions< T >::y().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::erode | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
std::vector< KeyPoint > fast_corners | ( | const SimpleTensor< T > & | src, |
float | input_thresh, | ||
bool | suppress_nonmax, | ||
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 152 of file FastCorners.cpp.
References arm_compute::test::coord2index(), KeyPoint::error, arm_compute::test::index2coord(), arm_compute::test::is_in_valid_region(), non_maxima_suppression(), SimpleTensor< T >::num_elements(), offset(), KeyPoint::orientation, KeyPoint::scale, SimpleTensor< T >::shape(), arm_compute::test::shape_to_valid_region(), KeyPoint::strength, arm_compute::test::validation::tensor_elem_at(), threshold(), KeyPoint::tracking_status, arm_compute::U8, arm_compute::UNDEFINED, arm_compute::test::validation::valid_region, Dimensions< T >::x(), KeyPoint::x, Dimensions< T >::y(), and KeyPoint::y.
template std::vector<KeyPoint> arm_compute::test::validation::reference::fast_corners | ( | const SimpleTensor< uint8_t > & | src, |
float | threshold, | ||
bool | suppress_nonmax, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > flatten_layer | ( | const SimpleTensor< T > & | src, |
const TensorShape & | shape_flatten | ||
) |
Definition at line 35 of file FlattenLayer.cpp.
References copy(), SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::flatten_layer | ( | const SimpleTensor< float > & | src, |
const TensorShape & | shape_flatten | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::flatten_layer | ( | const SimpleTensor< half > & | src, |
const TensorShape & | shape_flatten | ||
) |
SimpleTensor< T > floor_layer | ( | const SimpleTensor< T > & | src | ) |
Definition at line 39 of file Floor.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<half> arm_compute::test::validation::reference::floor_layer | ( | const SimpleTensor< half > & | src | ) |
template SimpleTensor<float> arm_compute::test::validation::reference::floor_layer | ( | const SimpleTensor< float > & | src | ) |
SimpleTensor< T > fully_connected_layer | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< T > & | weights, | ||
const SimpleTensor< TB > & | bias, | ||
const TensorShape & | dst_shape, | ||
QuantizationInfo | out_quant_info | ||
) |
Definition at line 115 of file FullyConnectedLayer.cpp.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, Dimensions< T >::num_dimensions(), SimpleTensor< T >::quantization_info(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and TensorShape::total_size_upper().
template SimpleTensor<float> arm_compute::test::validation::reference::fully_connected_layer | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | weights, | ||
const SimpleTensor< float > & | bias, | ||
const TensorShape & | dst_shape, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::fully_connected_layer | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | weights, | ||
const SimpleTensor< half > & | bias, | ||
const TensorShape & | dst_shape, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::fully_connected_layer | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint8_t > & | weights, | ||
const SimpleTensor< int32_t > & | bias, | ||
const TensorShape & | dst_shape, | ||
QuantizationInfo | out_quant_info | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::fully_connected_layer | ( | const SimpleTensor< int8_t > & | src, |
const SimpleTensor< int8_t > & | weights, | ||
const SimpleTensor< int32_t > & | bias, | ||
const TensorShape & | dst_shape, | ||
QuantizationInfo | out_quant_info | ||
) |
void fuse_batch_normalization_conv_layer | ( | const SimpleTensor< T > & | w, |
const SimpleTensor< T > & | mean, | ||
const SimpleTensor< T > & | var, | ||
SimpleTensor< T > & | w_fused, | ||
SimpleTensor< T > & | b_fused, | ||
const SimpleTensor< T > & | b, | ||
const SimpleTensor< T > & | beta, | ||
const SimpleTensor< T > & | gamma, | ||
float | epsilon | ||
) |
Definition at line 71 of file FuseBatchNormalization.cpp.
References arm_compute::test::validation::b, SimpleTensor< T >::data(), arm_compute::quantization::epsilon, fuse_batch_normalization_dwc_layer(), SimpleTensor< T >::shape(), and arm_compute::test::validation::w.
template void arm_compute::test::validation::reference::fuse_batch_normalization_conv_layer | ( | const SimpleTensor< float > & | w, |
const SimpleTensor< float > & | mean, | ||
const SimpleTensor< float > & | var, | ||
SimpleTensor< float > & | w_fused, | ||
SimpleTensor< float > & | b_fused, | ||
const SimpleTensor< float > & | b, | ||
const SimpleTensor< float > & | beta, | ||
const SimpleTensor< float > & | gamma, | ||
float | epsilon | ||
) |
template void arm_compute::test::validation::reference::fuse_batch_normalization_conv_layer | ( | const SimpleTensor< half > & | w, |
const SimpleTensor< half > & | mean, | ||
const SimpleTensor< half > & | var, | ||
SimpleTensor< half > & | w_fused, | ||
SimpleTensor< half > & | b_fused, | ||
const SimpleTensor< half > & | b, | ||
const SimpleTensor< half > & | beta, | ||
const SimpleTensor< half > & | gamma, | ||
float | epsilon | ||
) |
void fuse_batch_normalization_dwc_layer | ( | const SimpleTensor< T > & | w, |
const SimpleTensor< T > & | mean, | ||
const SimpleTensor< T > & | var, | ||
SimpleTensor< T > & | w_fused, | ||
SimpleTensor< T > & | b_fused, | ||
const SimpleTensor< T > & | b, | ||
const SimpleTensor< T > & | beta, | ||
const SimpleTensor< T > & | gamma, | ||
float | epsilon | ||
) |
Definition at line 36 of file FuseBatchNormalization.cpp.
References arm_compute::test::validation::b, SimpleTensor< T >::data(), and SimpleTensor< T >::shape().
Referenced by fuse_batch_normalization_conv_layer().
template void arm_compute::test::validation::reference::fuse_batch_normalization_dwc_layer | ( | const SimpleTensor< float > & | w, |
const SimpleTensor< float > & | mean, | ||
const SimpleTensor< float > & | var, | ||
SimpleTensor< float > & | w_fused, | ||
SimpleTensor< float > & | b_fused, | ||
const SimpleTensor< float > & | b, | ||
const SimpleTensor< float > & | beta, | ||
const SimpleTensor< float > & | gamma, | ||
float | epsilon | ||
) |
template void arm_compute::test::validation::reference::fuse_batch_normalization_dwc_layer | ( | const SimpleTensor< half > & | w, |
const SimpleTensor< half > & | mean, | ||
const SimpleTensor< half > & | var, | ||
SimpleTensor< half > & | w_fused, | ||
SimpleTensor< half > & | b_fused, | ||
const SimpleTensor< half > & | b, | ||
const SimpleTensor< half > & | beta, | ||
const SimpleTensor< half > & | gamma, | ||
float | epsilon | ||
) |
SimpleTensor< T > gather | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< uint32_t > & | indices, | ||
uint32_t | actual_axis | ||
) |
Definition at line 40 of file Gather.cpp.
References arm_compute::misc::shape_calculator::compute_gather_shape(), SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, arm_compute::execute_window_loop(), offset(), Dimensions< T >::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and Window::use_tensor_dimensions().
template SimpleTensor<float> arm_compute::test::validation::reference::gather | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< uint32_t > & | indices, | ||
uint32_t | actual_axis | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::gather | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< uint32_t > & | indices, | ||
uint32_t | actual_axis | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::gather | ( | const SimpleTensor< uint16_t > & | src, |
const SimpleTensor< uint32_t > & | indices, | ||
uint32_t | actual_axis | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::gather | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint32_t > & | indices, | ||
uint32_t | actual_axis | ||
) |
SimpleTensor< T > gaussian3x3 | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 38 of file Gaussian3x3.cpp.
References arm_compute::test::validation::apply_2d_spatial_filter(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), scale(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and arm_compute::utils::cast::U.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::gaussian3x3 | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > gaussian5x5 | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 38 of file Gaussian5x5.cpp.
References arm_compute::test::validation::apply_2d_spatial_filter(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), scale(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and arm_compute::utils::cast::U.
Referenced by gaussian_pyramid_half(), and laplacian_pyramid().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::gaussian5x5 | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
std::vector< SimpleTensor< T > > gaussian_pyramid_half | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
size_t | num_levels | ||
) |
Definition at line 41 of file GaussianPyramidHalf.cpp.
References arm_compute::CENTER, arm_compute::test::validation::dst, gaussian5x5(), arm_compute::NEAREST_NEIGHBOR, scale(), arm_compute::SCALE_PYRAMID_HALF, and arm_compute::test::validation::src.
Referenced by laplacian_pyramid(), and optical_flow().
template std::vector<SimpleTensor<uint8_t> > arm_compute::test::validation::reference::gaussian_pyramid_half | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
size_t | num_levels | ||
) |
SimpleTensor< T > gemm | ( | const SimpleTensor< T > & | a, |
const SimpleTensor< T > & | b, | ||
const SimpleTensor< T > & | c, | ||
float | alpha, | ||
float | beta | ||
) |
Definition at line 38 of file GEMM.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, K, M, N, SimpleTensor< T >::shape(), and arm_compute::test::validation::w.
Referenced by gemm_mixed_precision().
template SimpleTensor<float> arm_compute::test::validation::reference::gemm | ( | const SimpleTensor< float > & | a, |
const SimpleTensor< float > & | b, | ||
const SimpleTensor< float > & | c, | ||
float | alpha, | ||
float | beta | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::gemm | ( | const SimpleTensor< half > & | a, |
const SimpleTensor< half > & | b, | ||
const SimpleTensor< half > & | c, | ||
float | alpha, | ||
float | beta | ||
) |
SimpleTensor<T> arm_compute::test::validation::reference::gemm_interleave_4x4 | ( | const SimpleTensor< T > & | in, |
SimpleTensor< T > & | out | ||
) |
Definition at line 37 of file GEMMInterleave4x4.h.
References SimpleTensor< T >::data(), arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
SimpleTensor<T> arm_compute::test::validation::reference::gemm_interleave_blocked | ( | const SimpleTensor< T > & | in, |
SimpleTensor< T > & | out, | ||
int | int_by, | ||
int | block, | ||
bool | transposed | ||
) |
Definition at line 50 of file GEMMInterleaveBlocked.h.
References M, N, safe_read(), and SimpleTensor< T >::shape().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::gemm_interleave_blocked | ( | const SimpleTensor< uint8_t > & | in, |
SimpleTensor< uint8_t > & | out, | ||
int | int_by, | ||
int | block, | ||
bool | transposed | ||
) |
SimpleTensor< T > gemm_mixed_precision | ( | const SimpleTensor< T > & | a, |
const SimpleTensor< T > & | b, | ||
const SimpleTensor< T > & | c, | ||
float | alpha, | ||
float | beta | ||
) |
Definition at line 92 of file GEMM.cpp.
References arm_compute::test::validation::b, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, gemm(), K, M, N, SimpleTensor< T >::shape(), and arm_compute::test::validation::w.
template SimpleTensor<half> arm_compute::test::validation::reference::gemm_mixed_precision | ( | const SimpleTensor< half > & | a, |
const SimpleTensor< half > & | b, | ||
const SimpleTensor< half > & | c, | ||
float | alpha, | ||
float | beta | ||
) |
SimpleTensor< T > gemm_reshape_lhs_matrix | ( | const SimpleTensor< T > & | in, |
const TensorShape & | output_shape, | ||
const GEMMLHSMatrixInfo & | lhs_info | ||
) |
Definition at line 43 of file GEMMReshapeLHSMatrix.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::B, copy(), SimpleTensor< T >::data_type(), GEMMLHSMatrixInfo::interleave, K, GEMMLHSMatrixInfo::k0, M, GEMMLHSMatrixInfo::m0, arm_compute::test::validation::output_shape, SimpleTensor< T >::shape(), GEMMLHSMatrixInfo::transpose, and GEMMLHSMatrixInfo::v0.
template SimpleTensor<int> arm_compute::test::validation::reference::gemm_reshape_lhs_matrix | ( | const SimpleTensor< int > & | in, |
const TensorShape & | output_shape, | ||
const GEMMLHSMatrixInfo & | lhs_info | ||
) |
template SimpleTensor<short> arm_compute::test::validation::reference::gemm_reshape_lhs_matrix | ( | const SimpleTensor< short > & | in, |
const TensorShape & | output_shape, | ||
const GEMMLHSMatrixInfo & | lhs_info | ||
) |
template SimpleTensor<char> arm_compute::test::validation::reference::gemm_reshape_lhs_matrix | ( | const SimpleTensor< char > & | in, |
const TensorShape & | output_shape, | ||
const GEMMLHSMatrixInfo & | lhs_info | ||
) |
SimpleTensor< T > gemm_reshape_rhs_matrix | ( | const SimpleTensor< T > & | in, |
const TensorShape & | output_shape, | ||
const GEMMRHSMatrixInfo & | rhs_info | ||
) |
Definition at line 43 of file GEMMReshapeRHSMatrix.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::B, copy(), SimpleTensor< T >::data_type(), GEMMRHSMatrixInfo::h0, GEMMRHSMatrixInfo::interleave, K, GEMMRHSMatrixInfo::k0, N, GEMMRHSMatrixInfo::n0, arm_compute::test::validation::output_shape, SimpleTensor< T >::shape(), and GEMMRHSMatrixInfo::transpose.
template SimpleTensor<int> arm_compute::test::validation::reference::gemm_reshape_rhs_matrix | ( | const SimpleTensor< int > & | in, |
const TensorShape & | output_shape, | ||
const GEMMRHSMatrixInfo & | rhs_info | ||
) |
template SimpleTensor<short> arm_compute::test::validation::reference::gemm_reshape_rhs_matrix | ( | const SimpleTensor< short > & | in, |
const TensorShape & | output_shape, | ||
const GEMMRHSMatrixInfo & | rhs_info | ||
) |
template SimpleTensor<char> arm_compute::test::validation::reference::gemm_reshape_rhs_matrix | ( | const SimpleTensor< char > & | in, |
const TensorShape & | output_shape, | ||
const GEMMRHSMatrixInfo & | rhs_info | ||
) |
SimpleTensor<T> arm_compute::test::validation::reference::gemm_transpose_1xW | ( | const SimpleTensor< T > & | in | ) |
Definition at line 37 of file GEMMTranspose1xW.h.
References SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), and SimpleTensor< T >::shape().
SimpleTensor< T1 > gemmlowp | ( | const SimpleTensor< T2 > & | a, |
const SimpleTensor< T3 > & | b, | ||
TensorShape | shape_c | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::gemmlowp< int32_t, int8_t, int8_t > | ( | const SimpleTensor< int8_t > & | a, |
const SimpleTensor< int8_t > & | b, | ||
TensorShape | shape_c | ||
) |
Referenced by gemmlowp_quantize_down_scale_by_float().
template SimpleTensor<int32_t> arm_compute::test::validation::reference::gemmlowp< int32_t, uint8_t, int8_t > | ( | const SimpleTensor< uint8_t > & | a, |
const SimpleTensor< int8_t > & | b, | ||
TensorShape | shape_c | ||
) |
Referenced by gemmlowp_quantize_down_scale_by_float().
template SimpleTensor<int32_t> arm_compute::test::validation::reference::gemmlowp< int32_t, uint8_t, uint8_t > | ( | const SimpleTensor< uint8_t > & | a, |
const SimpleTensor< uint8_t > & | b, | ||
TensorShape | shape_c | ||
) |
Referenced by gemmlowp_quantize_down_scale_by_float().
SimpleTensor<T1> arm_compute::test::validation::reference::gemmlowp_matrix_multiply_core | ( | const SimpleTensor< T2 > & | a, |
const SimpleTensor< T3 > & | b, | ||
TensorShape | shape_c, | ||
int32_t | a_offset, | ||
int32_t | b_offset | ||
) |
SimpleTensor<T_out> arm_compute::test::validation::reference::gemmlowp_matrix_multiply_core | ( | const SimpleTensor< T_in > & | a, |
const SimpleTensor< T_in_1 > & | b, | ||
TensorShape | shape_c, | ||
int32_t | a_offset, | ||
int32_t | b_offset | ||
) |
Definition at line 181 of file GEMMLowp.cpp.
References type.
Referenced by gemmlowp_quantize_down_scale_by_float().
template SimpleTensor<int32_t> arm_compute::test::validation::reference::gemmlowp_matrix_multiply_core | ( | const SimpleTensor< int8_t > & | a, |
const SimpleTensor< int8_t > & | b, | ||
TensorShape | shape_c, | ||
int32_t | a_offset, | ||
int32_t | b_offset | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::gemmlowp_matrix_multiply_core | ( | const SimpleTensor< uint8_t > & | a, |
const SimpleTensor< uint8_t > & | b, | ||
TensorShape | shape_c, | ||
int32_t | a_offset, | ||
int32_t | b_offset | ||
) |
SimpleTensor<uint8_t> arm_compute::test::validation::reference::gemmlowp_quantize_down_scale | ( | const SimpleTensor< TIn > & | in, |
int32_t | result_offset, | ||
std::vector< int32_t > | result_mult_int, | ||
std::vector< int32_t > | result_shift | ||
) |
SimpleTensor< TOut > gemmlowp_quantize_down_scale | ( | const SimpleTensor< TIn > & | in, |
int32_t | result_offset, | ||
std::vector< int32_t > | result_mult_int, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
Definition at line 241 of file GEMMLowp.cpp.
References arm_compute::test::validation::data_type, arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
Referenced by gemmlowp_quantize_down_scale_by_float().
SimpleTensor< TOut > gemmlowp_quantize_down_scale | ( | const SimpleTensor< TIn > & | in, |
const SimpleTensor< TIn > & | bias, | ||
int32_t | result_offset, | ||
std::vector< int32_t > | result_mult_int, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
Definition at line 252 of file GEMMLowp.cpp.
References arm_compute::test::validation::data_type, arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
template SimpleTensor< int8_t > gemmlowp_quantize_down_scale | ( | const SimpleTensor< int32_t > & | a, |
int32_t | result_offset, | ||
std::vector< int32_t > | result_mult_int, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
template SimpleTensor< int8_t > gemmlowp_quantize_down_scale | ( | const SimpleTensor< int32_t > & | a, |
const SimpleTensor< int32_t > & | b, | ||
int32_t | result_offset, | ||
std::vector< int32_t > | result_mult_int, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
SimpleTensor< TOut > gemmlowp_quantize_down_scale_by_fixedpoint | ( | const SimpleTensor< TIn > & | in, |
std::vector< int32_t > | result_fixedpoint_multiplier, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | result_offset_after_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
Definition at line 263 of file GEMMLowp.cpp.
References arm_compute::test::validation::data_type, arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
Referenced by gemmlowp_quantize_down_scale_by_float().
SimpleTensor< TOut > gemmlowp_quantize_down_scale_by_fixedpoint | ( | const SimpleTensor< TIn > & | in, |
const SimpleTensor< TIn > & | bias, | ||
std::vector< int32_t > | result_fixedpoint_multiplier, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | result_offset_after_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
Definition at line 274 of file GEMMLowp.cpp.
References arm_compute::test::validation::data_type, arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
template SimpleTensor< int16_t > gemmlowp_quantize_down_scale_by_fixedpoint | ( | const SimpleTensor< int32_t > & | a, |
std::vector< int32_t > | result_fixedpoint_multiplier, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | result_offset_after_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
template SimpleTensor< int16_t > gemmlowp_quantize_down_scale_by_fixedpoint | ( | const SimpleTensor< int32_t > & | a, |
const SimpleTensor< int32_t > & | b, | ||
std::vector< int32_t > | result_fixedpoint_multiplier, | ||
std::vector< int32_t > | result_shift, | ||
int32_t | result_offset_after_shift, | ||
int32_t | min, | ||
int32_t | max | ||
) |
SimpleTensor< TOut > gemmlowp_quantize_down_scale_by_float | ( | const SimpleTensor< TIn > & | in, |
const SimpleTensor< TIn > & | bias, | ||
std::vector< float_t > | result_real_multiplier, | ||
int32_t | result_offset, | ||
int32_t | min, | ||
int32_t | max | ||
) |
Definition at line 285 of file GEMMLowp.cpp.
References arm_compute::test::validation::data_type, arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
Referenced by gemmlowp_quantize_down_scale_by_float().
SimpleTensor< TOut > gemmlowp_quantize_down_scale_by_float | ( | const SimpleTensor< TIn > & | in, |
std::vector< float_t > | result_real_multiplier, | ||
int32_t | result_offset, | ||
int32_t | min, | ||
int32_t | max | ||
) |
Definition at line 296 of file GEMMLowp.cpp.
References arm_compute::test::validation::b, arm_compute::test::validation::data_type, arm_compute::test::validation::dst, gemmlowp< int32_t, int8_t, int8_t >(), gemmlowp< int32_t, uint8_t, int8_t >(), gemmlowp< int32_t, uint8_t, uint8_t >(), gemmlowp_matrix_multiply_core(), gemmlowp_quantize_down_scale(), gemmlowp_quantize_down_scale_by_fixedpoint(), gemmlowp_quantize_down_scale_by_float(), and SimpleTensor< T >::shape().
template SimpleTensor< int8_t > gemmlowp_quantize_down_scale_by_float | ( | const SimpleTensor< int32_t > & | a, |
const SimpleTensor< int32_t > & | b, | ||
std::vector< float_t > | result_real_multiplier, | ||
int32_t | result_offset, | ||
int32_t | min, | ||
int32_t | max | ||
) |
template SimpleTensor< int8_t > gemmlowp_quantize_down_scale_by_float | ( | const SimpleTensor< int32_t > & | a, |
std::vector< float_t > | result_real_multiplier, | ||
int32_t | result_offset, | ||
int32_t | min, | ||
int32_t | max | ||
) |
std::vector< KeyPoint > harris_corner_detector | ( | const SimpleTensor< T > & | src, |
float | threshold, | ||
float | min_dist, | ||
float | sensitivity, | ||
int | gradient_size, | ||
int | block_size, | ||
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 187 of file HarrisCornerDetector.cpp.
References arm_compute::test::validation::src, and threshold().
template std::vector<KeyPoint> arm_compute::test::validation::reference::harris_corner_detector | ( | const SimpleTensor< uint8_t > & | src, |
float | threshold, | ||
float | min_dist, | ||
float | sensitivity, | ||
int | gradient_size, | ||
int | block_size, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< uint32_t > histogram | ( | const SimpleTensor< T > & | src, |
size_t | num_bins, | ||
int32_t | offset, | ||
uint32_t | range | ||
) |
Definition at line 38 of file Histogram.cpp.
References arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), offset(), range(), arm_compute::test::validation::src, and arm_compute::U32.
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::histogram | ( | const SimpleTensor< uint8_t > & | src, |
size_t | num_bins, | ||
int32_t | offset, | ||
uint32_t | range | ||
) |
void hog_block_normalization | ( | SimpleTensor< T > & | desc, |
const SimpleTensor< T > & | hog_space, | ||
const HOGInfo & | hog_info | ||
) |
Definition at line 174 of file HOGDescriptor.cpp.
References Size2D::area(), HOGInfo::block_size(), HOGInfo::block_stride(), HOGInfo::cell_size(), arm_compute::F32, Size2D::height, HOGInfo::num_bins(), HOGInfo::num_cells_per_block(), HOGInfo::num_cells_per_block_stride(), SimpleTensor< T >::shape(), and Size2D::width.
Referenced by hog_descriptor(), and hog_multi_detection().
template void arm_compute::test::validation::reference::hog_block_normalization | ( | SimpleTensor< float > & | desc, |
const SimpleTensor< float > & | hog_space, | ||
const HOGInfo & | hog_info | ||
) |
SimpleTensor< T > hog_descriptor | ( | const SimpleTensor< U > & | src, |
BorderMode | border_mode, | ||
U | constant_border_value, | ||
const HOGInfo & | hog_info | ||
) |
Definition at line 226 of file HOGDescriptor.cpp.
References HOGInfo::cell_size(), arm_compute::F32, arm_compute::GRAD_XY, Size2D::height, hog_block_normalization(), hog_orientation_binning(), arm_compute::L2NORM, magnitude(), HOGInfo::num_bins(), phase(), HOGInfo::phase_type(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and Size2D::width.
template SimpleTensor<float> arm_compute::test::validation::reference::hog_descriptor | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
const HOGInfo & | hog_info | ||
) |
std::vector< DetectionWindow > hog_detector | ( | const SimpleTensor< T > & | src, |
const std::vector< T > & | descriptor, | ||
unsigned int | max_num_detection_windows, | ||
const HOGInfo & | hog_info, | ||
const Size2D & | detection_window_stride, | ||
float | threshold, | ||
uint16_t | idx_class | ||
) |
Definition at line 48 of file HOGDetector.cpp.
References Size2D::area(), ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_MSG, arm_compute::test::validation::b, HOGInfo::block_stride(), HOGInfo::descriptor_size(), HOGInfo::detection_window_size(), Size2D::height, DetectionWindow::height, DetectionWindow::idx_class, HOGInfo::num_bins(), HOGInfo::num_block_positions_per_image(), HOGInfo::num_cells_per_block(), SimpleTensor< T >::num_channels(), DetectionWindow::score, SimpleTensor< T >::shape(), arm_compute::test::validation::src, threshold(), Size2D::width, DetectionWindow::width, DetectionWindow::x, and DetectionWindow::y.
Referenced by hog_multi_detection().
template std::vector<DetectionWindow> arm_compute::test::validation::reference::hog_detector | ( | const SimpleTensor< float > & | src, |
const std::vector< float > & | descriptor, | ||
unsigned int | max_num_detection_windows, | ||
const HOGInfo & | hog_info, | ||
const Size2D & | detection_window_stride, | ||
float | threshold, | ||
uint16_t | idx_class | ||
) |
std::vector< DetectionWindow > hog_multi_detection | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value, | ||
const std::vector< HOGInfo > & | models, | ||
std::vector< std::vector< float >> | descriptors, | ||
unsigned int | max_num_detection_windows, | ||
float | threshold, | ||
bool | non_maxima_suppression, | ||
float | min_distance | ||
) |
Definition at line 138 of file HOGMultiDetection.cpp.
References ARM_COMPUTE_ERROR_ON, detection_windows_non_maxima_suppression(), arm_compute::F32, arm_compute::GRAD_XY, Size2D::height, hog_block_normalization(), hog_detector(), hog_orientation_binning(), arm_compute::L2NORM, magnitude(), non_maxima_suppression(), TensorInfo::num_channels(), phase(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, TensorInfo::tensor_shape(), threshold(), and Size2D::width.
template std::vector<DetectionWindow> arm_compute::test::validation::reference::hog_multi_detection | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
const std::vector< HOGInfo > & | models, | ||
std::vector< std::vector< float >> | descriptors, | ||
unsigned int | max_num_detection_windows, | ||
float | threshold, | ||
bool | non_maxima_suppression, | ||
float | min_distance | ||
) |
void hog_orientation_binning | ( | const SimpleTensor< T > & | mag, |
const SimpleTensor< U > & | phase, | ||
SimpleTensor< V > & | hog_space, | ||
const HOGInfo & | hog_info | ||
) |
Definition at line 117 of file HOGDescriptor.cpp.
References HOGInfo::cell_size(), arm_compute::F32, Size2D::height, HOGInfo::num_bins(), SimpleTensor< T >::shape(), and Size2D::width.
Referenced by hog_descriptor(), and hog_multi_detection().
template void arm_compute::test::validation::reference::hog_orientation_binning | ( | const SimpleTensor< int16_t > & | mag, |
const SimpleTensor< uint8_t > & | phase, | ||
SimpleTensor< float > & | hog_space, | ||
const HOGInfo & | hog_info | ||
) |
void im2col | ( | const SimpleTensor< T > & | src, |
SimpleTensor< T > & | dst, | ||
const Size2D & | kernel_dims, | ||
const PadStrideInfo & | conv_info, | ||
bool | has_bias, | ||
unsigned int | num_groups | ||
) |
Definition at line 144 of file Im2Col.cpp.
References ARM_COMPUTE_ERROR, arm_compute::test::validation::conv_info, SimpleTensor< T >::data_layout(), arm_compute::test::validation::dst, arm_compute::test::validation::has_bias, im2col_nchw(), im2col_nhwc(), arm_compute::NCHW, arm_compute::NHWC, arm_compute::test::validation::num_groups, and arm_compute::test::validation::src.
template void arm_compute::test::validation::reference::im2col | ( | const SimpleTensor< uint8_t > & | src, |
SimpleTensor< uint8_t > & | dst, | ||
const Size2D & | kernel_dims, | ||
const PadStrideInfo & | conv_info, | ||
bool | has_bias, | ||
unsigned int | num_groups | ||
) |
template void arm_compute::test::validation::reference::im2col | ( | const SimpleTensor< half > & | src, |
SimpleTensor< half > & | dst, | ||
const Size2D & | kernel_dims, | ||
const PadStrideInfo & | conv_info, | ||
bool | has_bias, | ||
unsigned int | num_groups | ||
) |
template void arm_compute::test::validation::reference::im2col | ( | const SimpleTensor< float > & | src, |
SimpleTensor< float > & | dst, | ||
const Size2D & | kernel_dims, | ||
const PadStrideInfo & | conv_info, | ||
bool | has_bias, | ||
unsigned int | num_groups | ||
) |
void arm_compute::test::validation::reference::im2col_nchw | ( | const SimpleTensor< T > & | src, |
SimpleTensor< T > & | dst, | ||
const Size2D & | kernel_dims, | ||
const PadStrideInfo & | conv_info, | ||
bool | has_bias, | ||
unsigned int | num_groups | ||
) |
Definition at line 39 of file Im2Col.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::b, batches, arm_compute::CONSTANT, SimpleTensor< T >::data_layout(), SimpleTensor< T >::data_type(), Size2D::height, arm_compute::is_data_type_quantized_asymmetric(), arm_compute::NCHW, arm_compute::test::validation::num_groups, PadStrideInfo::pad(), SimpleTensor< T >::quantization_info(), arm_compute::scaled_dimensions(), SimpleTensor< T >::shape(), PadStrideInfo::stride(), arm_compute::test::validation::tensor_elem_at(), and Size2D::width.
Referenced by im2col().
void arm_compute::test::validation::reference::im2col_nhwc | ( | const SimpleTensor< T > & | src, |
SimpleTensor< T > & | dst, | ||
const Size2D & | kernel_dims, | ||
const PadStrideInfo & | conv_info, | ||
bool | has_bias | ||
) |
Definition at line 93 of file Im2Col.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::b, batches, arm_compute::CONSTANT, SimpleTensor< T >::data_layout(), SimpleTensor< T >::data_type(), Size2D::height, arm_compute::is_data_type_quantized_asymmetric(), arm_compute::NHWC, PadStrideInfo::pad(), SimpleTensor< T >::quantization_info(), arm_compute::scaled_dimensions(), SimpleTensor< T >::shape(), PadStrideInfo::stride(), arm_compute::test::validation::tensor_elem_at(), and Size2D::width.
Referenced by im2col().
SimpleTensor< T > instance_normalization | ( | const SimpleTensor< T > & | src, |
float | gamma, | ||
float | beta, | ||
float | epsilon | ||
) |
Definition at line 40 of file InstanceNormalizationLayer.cpp.
References arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::quantization::epsilon, SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::instance_normalization | ( | const SimpleTensor< float > & | src, |
float | gamma, | ||
float | beta, | ||
float | epsilon | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::instance_normalization | ( | const SimpleTensor< half > & | src, |
float | gamma, | ||
float | beta, | ||
float | epsilon | ||
) |
SimpleTensor< uint32_t > integral_image | ( | const SimpleTensor< T > & | src | ) |
Definition at line 35 of file IntegralImage.cpp.
References arm_compute::test::validation::dst, SimpleTensor< T >::shape(), arm_compute::test::validation::src, and arm_compute::U32.
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::integral_image | ( | const SimpleTensor< uint8_t > & | src | ) |
SimpleTensor< T > l2_normalize | ( | const SimpleTensor< T > & | src, |
unsigned int | axis, | ||
float | epsilon | ||
) |
Definition at line 51 of file L2NormalizeLayer.cpp.
References ARM_COMPUTE_ERROR, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::quantization::epsilon, SimpleTensor< T >::shape(), arm_compute::test::validation::src, sum(), and arm_compute::SUM_SQUARE.
template SimpleTensor<float> arm_compute::test::validation::reference::l2_normalize | ( | const SimpleTensor< float > & | src, |
unsigned int | axis, | ||
float | epsilon | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::l2_normalize | ( | const SimpleTensor< half > & | src, |
unsigned int | axis, | ||
float | epsilon | ||
) |
std::vector< SimpleTensor< U > > laplacian_pyramid | ( | const SimpleTensor< T > & | src, |
SimpleTensor< U > & | dst, | ||
size_t | num_levels, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
Definition at line 40 of file LaplacianPyramid.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, gaussian5x5(), gaussian_pyramid_half(), arm_compute::S16, arm_compute::test::validation::src, SUB, and arm_compute::WRAP.
template std::vector<SimpleTensor<int16_t> > arm_compute::test::validation::reference::laplacian_pyramid | ( | const SimpleTensor< uint8_t > & | src, |
SimpleTensor< int16_t > & | dst, | ||
size_t | num_levels, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< U > laplacian_reconstruct | ( | const std::vector< SimpleTensor< T >> & | pyramid, |
const SimpleTensor< T > & | low_res, | ||
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 40 of file LaplacianReconstruct.cpp.
References ADD, arithmetic_operation(), arm_compute::CENTER, arm_compute::test::validation::data_type, SimpleTensor< T >::data_type(), arm_compute::NEAREST_NEIGHBOR, arm_compute::SATURATE, scale(), arm_compute::test::validation::scale_x, arm_compute::test::validation::scale_y, arm_compute::test::validation::shape, and arm_compute::U8.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::laplacian_reconstruct | ( | const std::vector< SimpleTensor< int16_t >> & | pyramid, |
const SimpleTensor< int16_t > & | low_res, | ||
BorderMode | border_mode, | ||
int16_t | constant_border_value | ||
) |
SimpleTensor< T > logical_and | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2 | ||
) |
Definition at line 108 of file Logical.cpp.
References arm_compute::And, TensorShape::broadcast_shape(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
Referenced by logical_not().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::logical_and | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2 | ||
) |
T arm_compute::test::validation::reference::logical_binary_op | ( | arm_compute::LogicalOperation | op, |
T | src1, | ||
T | src2 | ||
) |
Definition at line 37 of file Logical.cpp.
References arm_compute::And, ARM_COMPUTE_ASSERT, arm_compute::test::coord2index(), arm_compute::test::validation::dst, arm_compute::Not, arm_compute::Or, Dimensions< T >::set(), SimpleTensor< T >::shape(), and arm_compute::Unknown.
SimpleTensor< T > logical_not | ( | const SimpleTensor< T > & | src | ) |
Definition at line 121 of file Logical.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, logical_and(), logical_or(), SimpleTensor< T >::num_elements(), and SimpleTensor< T >::shape().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::logical_not | ( | const SimpleTensor< uint8_t > & | src1 | ) |
SimpleTensor< T > logical_or | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2 | ||
) |
Definition at line 95 of file Logical.cpp.
References TensorShape::broadcast_shape(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::Or, and SimpleTensor< T >::shape().
Referenced by logical_not().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::logical_or | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2 | ||
) |
SimpleTensor< T > magnitude | ( | const SimpleTensor< T > & | gx, |
const SimpleTensor< T > & | gy, | ||
MagnitudeType | magnitude_type | ||
) |
Definition at line 35 of file Magnitude.cpp.
References SimpleTensor< T >::data_type(), arm_compute::L1NORM, SimpleTensor< T >::num_elements(), arm_compute::test::saturate_cast(), SimpleTensor< T >::shape(), and sum().
Referenced by NEMagnitudePhaseKernel< mag_type, phase_type >::configure(), CLGradientKernel::configure(), NEGradientKernel::configure(), CLMagnitudePhaseKernel::configure(), CLEdgeNonMaxSuppressionKernel::configure(), NEEdgeNonMaxSuppressionKernel::configure(), hog_descriptor(), hog_multi_detection(), magnitude_phase(), NEGradientKernel::name(), NEEdgeNonMaxSuppressionKernel::name(), NEGradientKernel::run(), and NEEdgeNonMaxSuppressionKernel::run().
template SimpleTensor<int16_t> arm_compute::test::validation::reference::magnitude | ( | const SimpleTensor< int16_t > & | gx, |
const SimpleTensor< int16_t > & | gy, | ||
MagnitudeType | magnitude_type | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::magnitude | ( | const SimpleTensor< int32_t > & | gx, |
const SimpleTensor< int32_t > & | gy, | ||
MagnitudeType | magnitude_type | ||
) |
template SimpleTensor<half_float::half> arm_compute::test::validation::reference::magnitude | ( | const SimpleTensor< half_float::half > & | gx, |
const SimpleTensor< half_float::half > & | gy, | ||
MagnitudeType | magnitude_type | ||
) |
SimpleTensor< T > max_unpooling_layer | ( | const SimpleTensor< T > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > & | indices, | ||
TensorShape | output_shape, | ||
DataLayout | data_layout | ||
) |
Definition at line 92 of file MaxUnpoolingLayer.cpp.
References arm_compute::test::validation::data_layout, arm_compute::test::validation::info, arm_compute::test::validation::output_shape, and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::max_unpooling_layer | ( | const SimpleTensor< float > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > & | indices, | ||
TensorShape | output_shape, | ||
DataLayout | data_layout | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::max_unpooling_layer | ( | const SimpleTensor< half > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > & | indices, | ||
TensorShape | output_shape, | ||
DataLayout | data_layout | ||
) |
SimpleTensor<uint8_t> arm_compute::test::validation::reference::max_unpooling_layer< uint8_t > | ( | const SimpleTensor< uint8_t > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > & | indices, | ||
TensorShape | output_shape, | ||
DataLayout | data_layout | ||
) |
Definition at line 79 of file MaxUnpoolingLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::data_layout, arm_compute::test::validation::dst, arm_compute::test::validation::info, and arm_compute::test::validation::output_shape.
SimpleTensor<T> arm_compute::test::validation::reference::max_unpooling_layer_internal | ( | const SimpleTensor< T > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > & | indices, | ||
TensorShape | output_shape, | ||
DataLayout | data_layout | ||
) |
Definition at line 41 of file MaxUnpoolingLayer.cpp.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, arm_compute::test::validation::b, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::output_shape, SimpleTensor< T >::shape(), and arm_compute::test::validation::w.
std::pair< float, float > mean_and_standard_deviation | ( | const SimpleTensor< T > & | in | ) |
Definition at line 35 of file MeanStdDev.cpp.
References accumulate(), arm_compute::test::validation::b, SimpleTensor< T >::data(), and SimpleTensor< T >::num_elements().
template std::pair<float, float> arm_compute::test::validation::reference::mean_and_standard_deviation | ( | const SimpleTensor< uint8_t > & | in | ) |
template std::pair<float, float> arm_compute::test::validation::reference::mean_and_standard_deviation | ( | const SimpleTensor< half > & | in | ) |
template std::pair<float, float> arm_compute::test::validation::reference::mean_and_standard_deviation | ( | const SimpleTensor< float > & | in | ) |
SimpleTensor< T > mean_std_normalization_layer | ( | const SimpleTensor< T > & | src, |
float | epsilon | ||
) |
Definition at line 37 of file MeanStdDevNormalizationLayer.cpp.
References caffe_mnist_image_extractor::cols, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::shape(), and sum().
template SimpleTensor<float> arm_compute::test::validation::reference::mean_std_normalization_layer | ( | const SimpleTensor< float > & | src, |
float | epsilon | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::mean_std_normalization_layer | ( | const SimpleTensor< half > & | src, |
float | epsilon | ||
) |
SimpleTensor< T > median3x3 | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 44 of file Median3x3.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), and Dimensions< T >::x().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::median3x3 | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
MinMaxLocationValues< T > min_max_location | ( | const SimpleTensor< T > & | src | ) |
Definition at line 58 of file MinMaxLocation.cpp.
References arm_compute::test::validation::dst, MinMaxLocationValues< MinMaxType >::max, MinMaxLocationValues< MinMaxType >::max_loc, MinMaxLocationValues< MinMaxType >::min, MinMaxLocationValues< MinMaxType >::min_loc, SimpleTensor< T >::shape(), arm_compute::test::validation::src, and Coordinates2D::x.
template MinMaxLocationValues<uint8_t> arm_compute::test::validation::reference::min_max_location | ( | const SimpleTensor< uint8_t > & | src | ) |
template MinMaxLocationValues<int16_t> arm_compute::test::validation::reference::min_max_location | ( | const SimpleTensor< int16_t > & | src | ) |
template MinMaxLocationValues<float> arm_compute::test::validation::reference::min_max_location | ( | const SimpleTensor< float > & | src | ) |
SimpleTensor< T > non_linear_filter | ( | const SimpleTensor< T > & | src, |
NonLinearFilterFunction | function, | ||
unsigned int | mask_size, | ||
MatrixPattern | pattern, | ||
const uint8_t * | mask, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
Definition at line 36 of file NonLinearFilter.cpp.
References ARM_COMPUTE_ERROR, ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), arm_compute::test::is_in_valid_region(), arm_compute::MAX, arm_compute::MEDIAN, arm_compute::MIN, SimpleTensor< T >::num_elements(), arm_compute::OTHER, arm_compute::test::saturate_cast(), SimpleTensor< T >::shape(), arm_compute::test::shape_to_valid_region(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), arm_compute::UNDEFINED, arm_compute::test::validation::valid_region, and Dimensions< T >::x().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::non_linear_filter | ( | const SimpleTensor< uint8_t > & | src, |
NonLinearFilterFunction | function, | ||
unsigned int | mask_size, | ||
MatrixPattern | pattern, | ||
const uint8_t * | mask, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< int > non_max_suppression | ( | const SimpleTensor< float > & | bboxes, |
const SimpleTensor< float > & | scores, | ||
SimpleTensor< int > & | indices, | ||
unsigned int | max_output_size, | ||
float | score_threshold, | ||
float | nms_threshold | ||
) |
Definition at line 139 of file NonMaxSuppression.cpp.
References SimpleTensor< T >::data(), arm_compute::test::validation::output_size, and SimpleTensor< T >::shape().
SimpleTensor< T > non_maxima_suppression | ( | const SimpleTensor< T > & | src, |
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 38 of file NonMaximaSuppression.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), arm_compute::test::is_in_valid_region(), SimpleTensor< T >::num_channels(), SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), arm_compute::test::shape_to_valid_region(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), arm_compute::UNDEFINED, arm_compute::test::validation::valid_region, Dimensions< T >::x(), and Dimensions< T >::y().
Referenced by NEHOGMultiDetection::configure(), CLHOGMultiDetection::configure(), fast_corners(), and hog_multi_detection().
template SimpleTensor<float> arm_compute::test::validation::reference::non_maxima_suppression | ( | const SimpleTensor< float > & | src, |
BorderMode | border_mode, | ||
float | constant_border_value | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::non_maxima_suppression | ( | const SimpleTensor< uint8_t > & | src, |
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > normalization_layer | ( | const SimpleTensor< T > & | src, |
NormalizationLayerInfo | info | ||
) |
Definition at line 37 of file NormalizationLayer.cpp.
References NormalizationLayerInfo::beta(), caffe_mnist_image_extractor::cols, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::IN_MAP_2D, NormalizationLayerInfo::is_cross_map(), NormalizationLayerInfo::kappa(), NormalizationLayerInfo::norm_size(), caffe_mnist_image_extractor::rows, NormalizationLayerInfo::scale_coeff(), SimpleTensor< T >::shape(), type, and NormalizationLayerInfo::type().
template SimpleTensor<float> arm_compute::test::validation::reference::normalization_layer | ( | const SimpleTensor< float > & | src, |
NormalizationLayerInfo | info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::normalization_layer | ( | const SimpleTensor< half > & | src, |
NormalizationLayerInfo | info | ||
) |
SimpleTensor< T > normalize_planar_yuv_layer | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< T > & | mean, | ||
const SimpleTensor< T > & | std | ||
) |
Definition at line 38 of file NormalizePlanarYUVLayer.cpp.
References caffe_mnist_image_extractor::cols, SimpleTensor< T >::data_type(), caffe_mnist_image_extractor::rows, and SimpleTensor< T >::shape().
Referenced by normalize_planar_yuv_layer< int8_t >().
template SimpleTensor<half> arm_compute::test::validation::reference::normalize_planar_yuv_layer | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | mean, | ||
const SimpleTensor< half > & | std | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::normalize_planar_yuv_layer | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | mean, | ||
const SimpleTensor< float > & | std | ||
) |
SimpleTensor<int8_t> arm_compute::test::validation::reference::normalize_planar_yuv_layer< int8_t > | ( | const SimpleTensor< int8_t > & | src, |
const SimpleTensor< int8_t > & | mean, | ||
const SimpleTensor< int8_t > & | std | ||
) |
Definition at line 76 of file NormalizePlanarYUVLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::dst, normalize_planar_yuv_layer(), and SimpleTensor< T >::quantization_info().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::normalize_planar_yuv_layer< uint8_t > | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint8_t > & | mean, | ||
const SimpleTensor< uint8_t > & | std | ||
) |
Definition at line 65 of file NormalizePlanarYUVLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::dst, and SimpleTensor< T >::quantization_info().
std::vector< KeyPoint > optical_flow | ( | const SimpleTensor< T > & | old_input, |
const SimpleTensor< T > & | new_input, | ||
const OpticalFlowParameters & | params, | ||
size_t | num_levels, | ||
const std::vector< KeyPoint > & | old_points, | ||
const std::vector< KeyPoint > & | new_points_estimates, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
Definition at line 246 of file OpticalFlow.cpp.
References OpticalFlowParameters::epsilon, gaussian_pyramid_half(), arm_compute::GRAD_XY, GemmTuner::level, OpticalFlowParameters::num_iterations, arm_compute::SCALE_PYRAMID_HALF, arm_compute::test::validation::shape, arm_compute::test::shape_to_valid_region(), arm_compute::TERM_CRITERIA_BOTH, arm_compute::TERM_CRITERIA_EPSILON, OpticalFlowParameters::termination, InternalKeyPoint::tracking_status, arm_compute::UNDEFINED, OpticalFlowParameters::use_initial_estimate, OpticalFlowParameters::window_dimension, InternalKeyPoint::x, and InternalKeyPoint::y.
template std::vector<KeyPoint> arm_compute::test::validation::reference::optical_flow | ( | const SimpleTensor< uint8_t > & | old_input, |
const SimpleTensor< uint8_t > & | new_input, | ||
const OpticalFlowParameters & | params, | ||
size_t | num_levels, | ||
const std::vector< KeyPoint > & | old_points, | ||
const std::vector< KeyPoint > & | new_points_estimates, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > pad_layer | ( | const SimpleTensor< T > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode = PaddingMode::CONSTANT |
||
) |
Reference function to pad an ND tensor.
This function is not supposed to be optimized, but to clearly and naively execute the padding of a tensor
[in] | src | Tensor to pad |
[in] | paddings | Padding size in each dimension |
[in] | const_value | Constant value to fill padding with |
[in] | mode | [optional] Padding mode to use |
Definition at line 39 of file PadLayer.cpp.
References ARM_COMPUTE_ERROR, arm_compute::misc::shape_calculator::compute_padded_shape(), arm_compute::CONSTANT, arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, PixelValue::get(), arm_compute::test::index2coord(), clang_tidy_rules::mode, SimpleTensor< T >::num_elements(), Dimensions< size_t >::num_max_dimensions, arm_compute::REFLECT, SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::SYMMETRIC, Dimensions< T >::x(), Dimensions< T >::y(), and Dimensions< T >::z().
Referenced by conv2d_dft().
template SimpleTensor<float> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< float > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< half > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< uint8_t > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< int8_t > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< uint16_t > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< int16_t > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< uint32_t > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::pad_layer | ( | const SimpleTensor< int32_t > & | src, |
const PaddingList & | paddings, | ||
const PixelValue | const_value = PixelValue() , |
||
const PaddingMode | mode | ||
) |
SimpleTensor< T > permute | ( | const SimpleTensor< T > & | src, |
PermutationVector | perm | ||
) |
Definition at line 38 of file Permute.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, arm_compute::test::index2coord(), SimpleTensor< T >::num_channels(), SimpleTensor< T >::num_elements(), SimpleTensor< T >::quantization_info(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
Referenced by dft_2d(), pooling_layer_internal(), rdft_2d(), and ridft_2d().
template SimpleTensor<int8_t> arm_compute::test::validation::reference::permute | ( | const SimpleTensor< int8_t > & | src, |
PermutationVector | perm | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::permute | ( | const SimpleTensor< uint8_t > & | src, |
PermutationVector | perm | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::permute | ( | const SimpleTensor< int16_t > & | src, |
PermutationVector | perm | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::permute | ( | const SimpleTensor< uint16_t > & | src, |
PermutationVector | perm | ||
) |
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::permute | ( | const SimpleTensor< uint32_t > & | src, |
PermutationVector | perm | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::permute | ( | const SimpleTensor< float > & | src, |
PermutationVector | perm | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::permute | ( | const SimpleTensor< half > & | src, |
PermutationVector | perm | ||
) |
SimpleTensor< uint8_t > phase | ( | const SimpleTensor< T > & | gx, |
const SimpleTensor< T > & | gy, | ||
PhaseType | phase_type | ||
) |
Definition at line 35 of file Phase.cpp.
References SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), arm_compute::U8, and arm_compute::UNSIGNED.
Referenced by NEMagnitudePhaseKernel< mag_type, phase_type >::configure(), NEGradientKernel::configure(), CLGradientKernel::configure(), CLMagnitudePhaseKernel::configure(), CLEdgeNonMaxSuppressionKernel::configure(), NEEdgeNonMaxSuppressionKernel::configure(), hog_descriptor(), hog_multi_detection(), magnitude_phase(), NEGradientKernel::name(), NEEdgeNonMaxSuppressionKernel::name(), NEHOGOrientationBinningKernel::run(), NEGradientKernel::run(), and NEEdgeNonMaxSuppressionKernel::run().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::phase | ( | const SimpleTensor< int16_t > & | gx, |
const SimpleTensor< int16_t > & | gy, | ||
PhaseType | phase_type | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::phase | ( | const SimpleTensor< int32_t > & | gx, |
const SimpleTensor< int32_t > & | gy, | ||
PhaseType | phase_type | ||
) |
SimpleTensor< T3 > pixel_wise_multiplication | ( | const SimpleTensor< T1 > & | src1, |
const SimpleTensor< T2 > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
Definition at line 217 of file PixelWiseMultiplication.cpp.
References ARM_COMPUTE_ERROR, ARM_COMPUTE_UNUSED, TensorShape::broadcast_shape(), arm_compute::test::validation::dst, and SimpleTensor< T >::shape().
Referenced by pixel_wise_multiplication().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::pixel_wise_multiplication | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
Definition at line 239 of file PixelWiseMultiplication.cpp.
References ARM_COMPUTE_ERROR, TensorShape::broadcast_shape(), arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::F32, arm_compute::QASYMM8, scale(), and SimpleTensor< T >::shape().
SimpleTensor<int16_t> arm_compute::test::validation::reference::pixel_wise_multiplication | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< uint8_t > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
Definition at line 267 of file PixelWiseMultiplication.cpp.
References ARM_COMPUTE_ERROR, TensorShape::broadcast_shape(), arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::F32, arm_compute::QASYMM8, scale(), and SimpleTensor< T >::shape().
SimpleTensor<int8_t> arm_compute::test::validation::reference::pixel_wise_multiplication | ( | const SimpleTensor< int8_t > & | src1, |
const SimpleTensor< int8_t > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
Definition at line 295 of file PixelWiseMultiplication.cpp.
References ARM_COMPUTE_ERROR, TensorShape::broadcast_shape(), arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::F32, arm_compute::QASYMM8_SIGNED, scale(), and SimpleTensor< T >::shape().
template SimpleTensor< int32_t > pixel_wise_multiplication | ( | const SimpleTensor< int16_t > & | src1, |
const SimpleTensor< int16_t > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
Definition at line 323 of file PixelWiseMultiplication.cpp.
References ARM_COMPUTE_ERROR, TensorShape::broadcast_shape(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::F32, pixel_wise_multiplication(), arm_compute::QSYMM16, scale(), and SimpleTensor< T >::shape().
template SimpleTensor<int16_t> arm_compute::test::validation::reference::pixel_wise_multiplication | ( | const SimpleTensor< uint8_t > & | src1, |
const SimpleTensor< int16_t > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::pixel_wise_multiplication | ( | const SimpleTensor< int32_t > & | src1, |
const SimpleTensor< int32_t > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::pixel_wise_multiplication | ( | const SimpleTensor< float > & | src1, |
const SimpleTensor< float > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
template SimpleTensor<half_float::half> arm_compute::test::validation::reference::pixel_wise_multiplication | ( | const SimpleTensor< half_float::half > & | src1, |
const SimpleTensor< half_float::half > & | src2, | ||
float | scale, | ||
ConvertPolicy | convert_policy, | ||
RoundingPolicy | rounding_policy, | ||
DataType | dt_out, | ||
const QuantizationInfo & | qout | ||
) |
SimpleTensor< T > pooling_layer | ( | const SimpleTensor< T > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Definition at line 184 of file PoolingLayer.cpp.
References ARM_COMPUTE_UNUSED, arm_compute::test::validation::data_layout, arm_compute::test::validation::info, and arm_compute::test::validation::src.
Referenced by pooling_layer().
SimpleTensor<half> arm_compute::test::validation::reference::pooling_layer | ( | const SimpleTensor< half > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Definition at line 210 of file PoolingLayer.cpp.
References ARM_COMPUTE_UNUSED, arm_compute::test::validation::data_layout, SimpleTensor< T >::data_type(), arm_compute::F16, PoolingLayerInfo::fp_mixed_precision, arm_compute::test::validation::info, pooling_layer(), pooling_layer_internal< half >(), pooling_layer_internal< half, float >(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::pooling_layer | ( | const SimpleTensor< float > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
SimpleTensor<int8_t> arm_compute::test::validation::reference::pooling_layer< int8_t > | ( | const SimpleTensor< int8_t > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Definition at line 201 of file PoolingLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::data_layout, arm_compute::test::validation::dst, arm_compute::test::validation::info, and pooling_layer_internal< float >().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::pooling_layer< uint8_t > | ( | const SimpleTensor< uint8_t > & | src, |
const PoolingLayerInfo & | info, | ||
const QuantizationInfo & | output_qinfo, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Definition at line 191 of file PoolingLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::data_layout, arm_compute::test::validation::dst, arm_compute::test::validation::info, and pooling_layer_internal< float >().
SimpleTensor< T > pooling_layer_internal | ( | const SimpleTensor< T > & | src, |
const PoolingLayerInfo & | info, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Definition at line 41 of file PoolingLayer.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::AVG, arm_compute::test::validation::b, arm_compute::misc::shape_calculator::compute_pool_shape(), arm_compute::test::coord2index(), arm_compute::test::validation::data_layout, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, PoolingLayerInfo::exclude_padding, Size2D::height, arm_compute::test::validation::info, PoolingLayerInfo::is_global_pooling, arm_compute::support::cpp11::lowest(), arm_compute::MAX, arm_compute::NCHW, PadStrideInfo::pad_bottom(), PadStrideInfo::pad_left(), PadStrideInfo::pad_right(), PoolingLayerInfo::pad_stride_info, PadStrideInfo::pad_top(), permute(), PoolingLayerInfo::pool_size, pool_stride_x, PoolingLayerInfo::pool_type, pooling_layer_internal< float >(), pooling_layer_internal< half >(), pooling_layer_internal< half, float >(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, PadStrideInfo::stride(), type, arm_compute::utils::cast::U, arm_compute::U32, arm_compute::test::validation::w, and Size2D::width.
template SimpleTensor<float> arm_compute::test::validation::reference::pooling_layer_internal< float > | ( | const SimpleTensor< float > & | src, |
const PoolingLayerInfo & | info, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Referenced by pooling_layer< int8_t >(), pooling_layer< uint8_t >(), and pooling_layer_internal().
template SimpleTensor<half> arm_compute::test::validation::reference::pooling_layer_internal< half > | ( | const SimpleTensor< half > & | src, |
const PoolingLayerInfo & | info, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Referenced by pooling_layer(), and pooling_layer_internal().
template SimpleTensor<half> arm_compute::test::validation::reference::pooling_layer_internal< half, float > | ( | const SimpleTensor< half > & | src, |
const PoolingLayerInfo & | info, | ||
SimpleTensor< uint32_t > * | indices, | ||
DataLayout | data_layout | ||
) |
Referenced by pooling_layer(), and pooling_layer_internal().
SimpleTensor< T > prior_box_layer | ( | const SimpleTensor< T > & | src1, |
const SimpleTensor< T > & | src2, | ||
const PriorBoxLayerInfo & | info, | ||
const TensorShape & | output_shape | ||
) |
Definition at line 39 of file PriorBoxLayer.cpp.
References PriorBoxLayerInfo::aspect_ratios(), PriorBoxLayerInfo::clip(), SimpleTensor< T >::data_type(), PriorBoxLayerInfo::img_size(), arm_compute::test::validation::info, PriorBoxLayerInfo::max_sizes(), PriorBoxLayerInfo::min_sizes(), PriorBoxLayerInfo::offset(), arm_compute::test::validation::output_shape, SimpleTensor< T >::shape(), PriorBoxLayerInfo::steps(), PriorBoxLayerInfo::variances(), arm_compute::test::validation::w, Coordinates2D::x, and Coordinates2D::y.
template SimpleTensor<float> arm_compute::test::validation::reference::prior_box_layer | ( | const SimpleTensor< float > & | src1, |
const SimpleTensor< float > & | src2, | ||
const PriorBoxLayerInfo & | info, | ||
const TensorShape & | output_shape | ||
) |
SimpleTensor< int16_t > qlstm_layer_normalization | ( | const SimpleTensor< int16_t > & | src, |
const SimpleTensor< int16_t > & | weight, | ||
const SimpleTensor< int32_t > & | bias | ||
) |
Definition at line 40 of file QLSTMLayerNormalization.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::quantization::calculate_quantized_multiplier(), arm_compute::quantization::get_invsqrt_quantized_multiplier_exp(), arm_compute::quantization::multiply_by_quantized_multiplier(), arm_compute::QSYMM16, SimpleTensor< T >::quantization_info(), SimpleTensor< T >::shape(), and sum().
Referenced by get_invsqrt_quantized_multiplier_exp().
SimpleTensor< Tout > quantization_layer | ( | const SimpleTensor< Tin > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
Definition at line 37 of file QuantizationLayer.cpp.
References ARM_COMPUTE_ERROR, arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), arm_compute::QASYMM16, arm_compute::QASYMM8, arm_compute::QASYMM8_SIGNED, arm_compute::test::validation::qinfo, arm_compute::quantize_qasymm16(), arm_compute::quantize_qasymm8(), arm_compute::quantize_qasymm8_signed(), SimpleTensor< T >::shape(), arm_compute::TO_NEAREST_EVEN, arm_compute::TO_ZERO, and QuantizationInfo::uniform().
Referenced by quantization_layer().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::quantization_layer | ( | const SimpleTensor< uint8_t > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
SimpleTensor<int8_t> arm_compute::test::validation::reference::quantization_layer | ( | const SimpleTensor< uint8_t > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
SimpleTensor<uint8_t> arm_compute::test::validation::reference::quantization_layer | ( | const SimpleTensor< int8_t > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
Definition at line 104 of file QuantizationLayer.cpp.
References arm_compute::test::validation::src.
SimpleTensor<int8_t> arm_compute::test::validation::reference::quantization_layer | ( | const SimpleTensor< int8_t > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
Definition at line 111 of file QuantizationLayer.cpp.
References arm_compute::test::validation::src.
SimpleTensor<uint16_t> arm_compute::test::validation::reference::quantization_layer | ( | const SimpleTensor< uint8_t > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
Definition at line 118 of file QuantizationLayer.cpp.
References quantization_layer(), and arm_compute::test::validation::src.
template SimpleTensor< uint16_t > quantization_layer | ( | const SimpleTensor< half > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
template SimpleTensor< uint16_t > quantization_layer | ( | const SimpleTensor< float > & | src, |
DataType | output_data_type, | ||
const QuantizationInfo & | quantization_info | ||
) |
SimpleTensor< T > range | ( | SimpleTensor< T > & | dst, |
float | start, | ||
const size_t | num_of_elements, | ||
float | step | ||
) |
Definition at line 50 of file Range.cpp.
References arm_compute::test::validation::dst.
Referenced by arm_compute::helpers::tensor_transform::compute_strided_slice_output_shape(), CLHistogramKernel::configure(), CLHistogramBorderKernel::configure(), clang_tidy_rules::filter_clang_tidy_lines(), histogram(), TestFilter::is_selected(), TFPreproccessor::preprocess(), format_doxygen::process_comment(), and range().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::range | ( | SimpleTensor< uint8_t > & | dst, |
float | start, | ||
const size_t | num_of_elements, | ||
float | step | ||
) |
Definition at line 57 of file Range.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::F32, arm_compute::QASYMM8, SimpleTensor< T >::quantization_info(), range(), SimpleTensor< T >::shape(), enable_tracing::start, and arm_compute::cpu::step.
template SimpleTensor<float> arm_compute::test::validation::reference::range | ( | SimpleTensor< float > & | dst, |
float | start, | ||
const size_t | num_of_elements, | ||
float | step | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::range | ( | SimpleTensor< half > & | dst, |
float | start, | ||
const size_t | num_of_elements, | ||
float | step | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::range | ( | SimpleTensor< int8_t > & | dst, |
float | start, | ||
const size_t | num_of_elements, | ||
float | step | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::range | ( | SimpleTensor< uint16_t > & | dst, |
float | start, | ||
const size_t | num_of_elements, | ||
float | step | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::range | ( | SimpleTensor< int16_t > & | dst, |
float | start, | ||
const size_t | num_of_elements, | ||
float | step | ||
) |
SimpleTensor< T > rdft_1d | ( | const SimpleTensor< T > & | src | ) |
Performs an one dimensional DFT on a real input.
[in] | src | Source tensor. |
Definition at line 313 of file DFT.cpp.
References Forward.
Referenced by conv2d_dft(), and arm_compute::test::validation::DATA_TEST_CASE().
template SimpleTensor<float> arm_compute::test::validation::reference::rdft_1d | ( | const SimpleTensor< float > & | src | ) |
template SimpleTensor<half> arm_compute::test::validation::reference::rdft_1d | ( | const SimpleTensor< half > & | src | ) |
SimpleTensor< T > rdft_2d | ( | const SimpleTensor< T > & | src | ) |
Performs a two dimensional DFT on a real input.
[in] | src | Source tensor. |
Definition at line 342 of file DFT.cpp.
References ARM_COMPUTE_ERROR_ON, Forward, SimpleTensor< T >::num_channels(), permute(), and arm_compute::utils::cast::U.
Referenced by conv2d_dft().
template SimpleTensor<float> arm_compute::test::validation::reference::rdft_2d | ( | const SimpleTensor< float > & | src | ) |
template SimpleTensor<half> arm_compute::test::validation::reference::rdft_2d | ( | const SimpleTensor< half > & | src | ) |
SimpleTensor< OT > reduction_operation | ( | const SimpleTensor< T > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | axis, | ||
ReductionOperation | op, | ||
QuantizationInfo | quantization_info_output | ||
) |
Definition at line 272 of file ReductionOperation.cpp.
References ARM_COMPUTE_UNUSED, arm_compute::test::validation::dst_shape, and arm_compute::test::validation::src.
Referenced by reduction_operation().
template SimpleTensor< int32_t > reduction_operation | ( | const SimpleTensor< uint8_t > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | axis, | ||
ReductionOperation | op, | ||
QuantizationInfo | quantization_info_output | ||
) |
Definition at line 279 of file ReductionOperation.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst_shape, arm_compute::MEAN_SUM, arm_compute::QASYMM8, SimpleTensor< T >::quantization_info(), and arm_compute::test::validation::src.
template SimpleTensor< int32_t > reduction_operation | ( | const SimpleTensor< int8_t > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | axis, | ||
ReductionOperation | op, | ||
QuantizationInfo | quantization_info_output | ||
) |
Definition at line 302 of file ReductionOperation.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst_shape, arm_compute::MEAN_SUM, arm_compute::QASYMM8_SIGNED, SimpleTensor< T >::quantization_info(), reduction_operation(), and arm_compute::test::validation::src.
template SimpleTensor< int32_t > reduction_operation | ( | const SimpleTensor< float > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | axis, | ||
ReductionOperation | op, | ||
QuantizationInfo | quantization_info_output = QuantizationInfo() |
||
) |
template SimpleTensor< int32_t > reduction_operation | ( | const SimpleTensor< half > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | axis, | ||
ReductionOperation | op, | ||
QuantizationInfo | quantization_info_output = QuantizationInfo() |
||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::reduction_operation | ( | const SimpleTensor< int32_t > & | src, |
const TensorShape & | dst_shape, | ||
unsigned int | axis, | ||
ReductionOperation | op, | ||
QuantizationInfo | quantization_info_output = QuantizationInfo() |
||
) |
SimpleTensor< T > remap | ( | const SimpleTensor< T > & | in, |
SimpleTensor< float > & | map_x, | ||
SimpleTensor< float > & | map_y, | ||
SimpleTensor< T > & | valid_mask, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
T | constant_border_value | ||
) |
Definition at line 41 of file Remap.cpp.
References arm_compute::AREA, ARM_COMPUTE_ERROR, ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_MSG, arm_compute::BILINEAR, arm_compute::test::validation::bilinear_policy(), SimpleTensor< T >::data_type(), arm_compute::test::index2coord(), arm_compute::NEAREST_NEIGHBOR, SimpleTensor< T >::num_elements(), arm_compute::REPLICATE, Dimensions< T >::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), arm_compute::UNDEFINED, and valid_bilinear_policy().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::remap | ( | const SimpleTensor< uint8_t > & | src, |
SimpleTensor< float > & | map_x, | ||
SimpleTensor< float > & | map_y, | ||
SimpleTensor< uint8_t > & | valid_mask, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > reorg_layer | ( | const SimpleTensor< T > & | src, |
int32_t | stride | ||
) |
Definition at line 38 of file ReorgLayer.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::b, arm_compute::misc::shape_calculator::compute_reorg_output_shape(), SimpleTensor< T >::data_layout(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::input_info, arm_compute::NCHW, offset(), arm_compute::test::validation::output_shape, Dimensions< T >::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and arm_compute::test::validation::w.
template SimpleTensor<int32_t> arm_compute::test::validation::reference::reorg_layer | ( | const SimpleTensor< int32_t > & | src, |
int32_t | stride | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::reorg_layer | ( | const SimpleTensor< int16_t > & | src, |
int32_t | stride | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::reorg_layer | ( | const SimpleTensor< int8_t > & | src, |
int32_t | stride | ||
) |
SimpleTensor< T > reshape_layer | ( | const SimpleTensor< T > & | src, |
const TensorShape & | output_shape | ||
) |
[ReshapeLayer]
Definition at line 38 of file ReshapeLayer.cpp.
References ARM_COMPUTE_ERROR_ON, SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), arm_compute::test::validation::output_shape, SimpleTensor< T >::shape(), arm_compute::test::validation::src, and TensorShape::total_size().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< uint8_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< int8_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< uint16_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< int16_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< uint32_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< int32_t > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< half > & | src, |
const TensorShape & | output_shape | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::reshape_layer | ( | const SimpleTensor< float > & | src, |
const TensorShape & | output_shape | ||
) |
SimpleTensor< T > reverse | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< uint32_t > & | axis | ||
) |
Definition at line 38 of file Reverse.cpp.
References ARM_COMPUTE_ERROR_ON, batches, arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_channels(), SimpleTensor< T >::num_elements(), SimpleTensor< T >::quantization_info(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
Referenced by conv2d_dft(), and arm_compute::utils::parse_npy_header().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::reverse | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint32_t > & | axis | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::reverse | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< uint32_t > & | axis | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::reverse | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< uint32_t > & | axis | ||
) |
SimpleTensor< T > ridft_1d | ( | const SimpleTensor< T > & | src, |
bool | is_odd = false |
||
) |
Performs an one dimensional inverse DFT on a real input.
[in] | src | Source tensor. |
[in] | is_odd | (Optional) Specifies if the output has odd dimensions. Is used by the inverse variant to reconstruct odd sequences. |
Definition at line 319 of file DFT.cpp.
References arm_compute::test::validation::dst, Inverse, and scale().
Referenced by conv2d_dft(), and arm_compute::test::validation::DATA_TEST_CASE().
template SimpleTensor<float> arm_compute::test::validation::reference::ridft_1d | ( | const SimpleTensor< float > & | src, |
bool | is_odd | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::ridft_1d | ( | const SimpleTensor< half > & | src, |
bool | is_odd | ||
) |
SimpleTensor< T > ridft_2d | ( | const SimpleTensor< T > & | src, |
bool | is_odd = false |
||
) |
Performs a two dimensional inverse DFT on a real input.
[in] | src | Source tensor. |
[in] | is_odd | (Optional) Specifies if the output has odd dimensions across width. Is used by the inverse variant to reconstruct odd sequences. |
Definition at line 354 of file DFT.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::dst, Inverse, SimpleTensor< T >::num_channels(), permute(), scale(), and arm_compute::utils::cast::U.
Referenced by conv2d_dft().
template SimpleTensor<float> arm_compute::test::validation::reference::ridft_2d | ( | const SimpleTensor< float > & | src, |
bool | is_odd | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::ridft_2d | ( | const SimpleTensor< half > & | src, |
bool | is_odd | ||
) |
SimpleTensor<T> arm_compute::test::validation::reference::roi_align_layer | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< TRois > & | rois, | ||
const ROIPoolingLayerInfo & | pool_info, | ||
const QuantizationInfo & | output_qinfo | ||
) |
SimpleTensor<float> arm_compute::test::validation::reference::roi_align_layer | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | rois, | ||
const ROIPoolingLayerInfo & | pool_info, | ||
const QuantizationInfo & | output_qinfo | ||
) |
Definition at line 140 of file ROIAlignLayer.cpp.
References ARM_COMPUTE_UNUSED, arm_compute::utility::clamp(), SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::output_shape, ROIPoolingLayerInfo::pooled_height(), ROIPoolingLayerInfo::pooled_width(), arm_compute::roi_align_1x1(), ROIPoolingLayerInfo::sampling_ratio(), SimpleTensor< T >::shape(), and ROIPoolingLayerInfo::spatial_scale().
SimpleTensor<half> arm_compute::test::validation::reference::roi_align_layer | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | rois, | ||
const ROIPoolingLayerInfo & | pool_info, | ||
const QuantizationInfo & | output_qinfo | ||
) |
Definition at line 213 of file ROIAlignLayer.cpp.
References arm_compute::test::validation::dst, arm_compute::F16, arm_compute::F32, and arm_compute::test::validation::src.
SimpleTensor<uint8_t> arm_compute::test::validation::reference::roi_align_layer | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint16_t > & | rois, | ||
const ROIPoolingLayerInfo & | pool_info, | ||
const QuantizationInfo & | output_qinfo | ||
) |
Definition at line 223 of file ROIAlignLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), and arm_compute::test::validation::dst.
SimpleTensor<int8_t> arm_compute::test::validation::reference::roi_align_layer | ( | const SimpleTensor< int8_t > & | src, |
const SimpleTensor< uint16_t > & | rois, | ||
const ROIPoolingLayerInfo & | pool_info, | ||
const QuantizationInfo & | output_qinfo | ||
) |
Definition at line 232 of file ROIAlignLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), and arm_compute::test::validation::dst.
T arm_compute::test::validation::reference::safe_read | ( | const SimpleTensor< T > & | t, |
int | y, | ||
int | x | ||
) |
Definition at line 37 of file GEMMInterleaveBlocked.h.
References M, N, and SimpleTensor< T >::shape().
Referenced by gemm_interleave_blocked().
SimpleTensor< T > scale | ( | const SimpleTensor< T > & | src, |
float | scale_x, | ||
float | scale_y, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
T | constant_border_value, | ||
SamplingPolicy | sampling_policy, | ||
bool | ceil_policy_scale, | ||
bool | align_corners | ||
) |
Definition at line 185 of file Scale.cpp.
References arm_compute::test::validation::sampling_policy, arm_compute::test::validation::scale_x, arm_compute::test::validation::scale_y, and arm_compute::test::validation::src.
Referenced by box3x3(), convolution(), dft_1d(), dft_2d(), gaussian3x3(), gaussian5x5(), gaussian_pyramid_half(), laplacian_reconstruct(), pixel_wise_multiplication(), ridft_1d(), ridft_2d(), and scale().
SimpleTensor<uint8_t> arm_compute::test::validation::reference::scale | ( | const SimpleTensor< uint8_t > & | src, |
float | scale_x, | ||
float | scale_y, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
SamplingPolicy | sampling_policy, | ||
bool | ceil_policy_scale, | ||
bool | align_corners | ||
) |
Definition at line 192 of file Scale.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::dequantize_qasymm8(), arm_compute::test::validation::dst, SimpleTensor< T >::quantization_info(), arm_compute::test::validation::sampling_policy, UniformQuantizationInfo::scale, arm_compute::test::validation::scale_x, arm_compute::test::validation::scale_y, arm_compute::test::validation::src, and QuantizationInfo::uniform().
SimpleTensor<int8_t> arm_compute::test::validation::reference::scale | ( | const SimpleTensor< int8_t > & | src, |
float | scale_x, | ||
float | scale_y, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
int8_t | constant_border_value, | ||
SamplingPolicy | sampling_policy, | ||
bool | ceil_policy_scale, | ||
bool | align_corners | ||
) |
Definition at line 211 of file Scale.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::dequantize_qasymm8_signed(), arm_compute::test::validation::dst, SimpleTensor< T >::quantization_info(), arm_compute::test::validation::sampling_policy, scale(), arm_compute::test::validation::scale_x, arm_compute::test::validation::scale_y, and arm_compute::test::validation::src.
template SimpleTensor<int16_t> arm_compute::test::validation::reference::scale | ( | const SimpleTensor< int16_t > & | src, |
float | scale_x, | ||
float | scale_y, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
int16_t | constant_border_value, | ||
SamplingPolicy | sampling_policy, | ||
bool | ceil_policy_scale, | ||
bool | align_corners | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::scale | ( | const SimpleTensor< half > & | src, |
float | scale_x, | ||
float | scale_y, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
half | constant_border_value, | ||
SamplingPolicy | sampling_policy, | ||
bool | ceil_policy_scale, | ||
bool | align_corners | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::scale | ( | const SimpleTensor< float > & | src, |
float | scale_x, | ||
float | scale_y, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
float | constant_border_value, | ||
SamplingPolicy | sampling_policy, | ||
bool | ceil_policy_scale, | ||
bool | align_corners | ||
) |
SimpleTensor<T> arm_compute::test::validation::reference::scale_core | ( | const SimpleTensor< T > & | in, |
float | scale_x, | ||
float | scale_y, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
T | constant_border_value, | ||
SamplingPolicy | sampling_policy, | ||
bool | ceil_policy_scale, | ||
bool | align_corners | ||
) |
Definition at line 41 of file Scale.cpp.
References arm_compute::AREA, ARM_COMPUTE_ERROR, ARM_COMPUTE_ERROR_ON, arm_compute::BILINEAR, arm_compute::test::validation::bilinear_policy(), arm_compute::scale_utils::calculate_resize_ratio(), arm_compute::CENTER, arm_compute::CONSTANT, arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::index2coord(), arm_compute::test::validation::is_valid_pixel_index(), arm_compute::NEAREST_NEIGHBOR, arm_compute::REPLICATE, arm_compute::utils::rounding::round_half_away_from_zero(), TensorShape::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::shape_scaled, sum(), arm_compute::test::validation::tensor_elem_at(), arm_compute::TOP_LEFT, arm_compute::utils::cast::U, arm_compute::UNDEFINED, and Dimensions< T >::x().
std::pair< SimpleTensor< T >, SimpleTensor< T > > scharr | ( | const SimpleTensor< U > & | src, |
int | filter_size, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
GradientDimension | gradient_dimension | ||
) |
Definition at line 62 of file Scharr.cpp.
References SimpleTensor< T >::shape().
template std::pair<SimpleTensor<int16_t>, SimpleTensor<int16_t> > arm_compute::test::validation::reference::scharr | ( | const SimpleTensor< uint8_t > & | src, |
int | filter_size, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
GradientDimension | gradient_dimension | ||
) |
SimpleTensor< T > select | ( | const SimpleTensor< uint8_t > & | c, |
const SimpleTensor< T > & | x, | ||
const SimpleTensor< T > & | y | ||
) |
Definition at line 38 of file Select.cpp.
References ARM_COMPUTE_ERROR_ON, SimpleTensor< T >::data(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, Dimensions< T >::num_dimensions(), SimpleTensor< T >::num_elements(), offset(), SimpleTensor< T >::shape(), and Dimensions< T >::x().
Referenced by combine_gradients_L1(), combine_gradients_L2(), harris_score_3x3(), harris_score_5x5(), harris_score_7x7(), hist_local_kernel(), minmax(), non_max_suppression(), phase_signed(), phase_unsigned(), pooling_layer_2_nchw_indices_fp16(), pooling_layer_2_nchw_indices_fp32(), product(), scale_bilinear_nchw(), scale_bilinear_quantized_nchw(), tablelookup_S16(), threshold_binary(), and threshold_range().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::select | ( | const SimpleTensor< uint8_t > & | c, |
const SimpleTensor< uint8_t > & | x, | ||
const SimpleTensor< uint8_t > & | y | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::select | ( | const SimpleTensor< uint8_t > & | c, |
const SimpleTensor< half > & | x, | ||
const SimpleTensor< half > & | y | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::select | ( | const SimpleTensor< uint8_t > & | c, |
const SimpleTensor< float > & | x, | ||
const SimpleTensor< float > & | y | ||
) |
SimpleTensor< T > slice | ( | const SimpleTensor< T > & | src, |
Coordinates | starts, | ||
Coordinates | ends | ||
) |
Definition at line 38 of file SliceOperations.cpp.
References ARM_COMPUTE_ERROR_ON, Dimensions< T >::cbegin(), arm_compute::misc::shape_calculator::compute_slice_shape(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, arm_compute::execute_window_loop(), Dimensions< T >::num_dimensions(), offset(), Dimensions< T >::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and Window::use_tensor_dimensions().
Referenced by Window::broadcast_if_dimension_le_one(), NEUnstack::configure(), CLUnstack::configure(), conv2d_dft(), deconvolution_layer(), depthconcatenate_layer(), IGCSimple2DKernel::run(), ICLSimple2DKernel::run(), ICLSimple3DKernel::run(), IGCSimple3DKernel::run(), GCScaleKernel::run(), GCTransposeKernel::run(), GCPixelWiseMultiplicationKernel::run(), GCNormalizationLayerKernel::run(), GCGEMMMatrixAdditionKernel::run(), GCAbsoluteDifferenceKernel::run(), CLBitwiseKernel::run(), GCActivationLayerKernel::run(), CLHistogramKernel::run(), CLMinMaxKernel::run(), GCDropoutLayerKernel::run(), CLGradientKernel::run(), GCLogits1DShiftExpSumKernel::run(), GCDirectConvolutionLayerKernel< kernel_size >::run(), GCFillBorderKernel::run(), CLAbsoluteDifferenceKernel::run(), CLRemapKernel::run(), CLHOGOrientationBinningKernel::run(), CLSobel3x3Kernel::run(), CLDequantizationLayerKernel::run(), CLDerivativeKernel::run(), CLSobel5x5HorKernel::run(), CLSobel7x7HorKernel::run(), GCPoolingLayerKernel::run(), GCArithmeticAdditionKernel::run(), GCNormalizePlanarYUVLayerKernel::run(), CLReverseKernel::run(), CLSelectKernel::run(), GCTensorShiftKernel::run(), CLMaxUnpoolingLayerKernel::run(), CLFFTScaleKernel::run(), CLQLSTMLayerNormalizationKernel::run(), CLQuantizationLayerKernel::run(), CLScaleKernel::run(), CLMagnitudePhaseKernel::run(), CLComparisonKernel::run(), CLIntegralImageVertKernel::run(), CLMinMaxLayerKernel::run(), CLNormalizationLayerKernel::run(), CLGEMMLowpQuantizeDownInt32ScaleByFixedPointKernel::run(), CLTileKernel::run(), CLFastCornersKernel::run(), CLFFTDigitReverseKernel::run(), CLMeanStdDevNormalizationKernel::run(), CLReorgLayerKernel::run(), CLROIPoolingLayerKernel::run(), GCCol2ImKernel::run(), GCGEMMMatrixMultiplyKernel::run(), GCIm2ColKernel::run(), CLNormalizePlanarYUVLayerKernel::run(), CLMeanStdDevKernel::run(), CLChannelExtractKernel::run(), CLPriorBoxLayerKernel::run(), CLScharr3x3Kernel::run(), GCBatchNormalizationLayerKernel::run(), CLPadLayerKernel::run(), CLHarrisScoreKernel::run(), CLHOGDetectorKernel::run(), CLFillBorderKernel::run(), CLFFTRadixStageKernel::run(), CLBoundingBoxTransformKernel::run(), CLChannelCombineKernel::run(), CLGEMMLowpQuantizeDownInt32ScaleByFloatKernel::run(), CLGEMMLowpQuantizeDownInt32ScaleKernel::run(), CLLogits1DMaxShiftExpSumKernel::run(), CLGEMMLowpMatrixMultiplyNativeKernel::run(), CLGEMMReshapeLHSMatrixKernel::run(), CLCol2ImKernel::run(), CLROIAlignLayerKernel::run(), GCLogits1DNormKernel::run(), CLHistogramBorderKernel::run(), CLDepthwiseConvolutionLayer3x3NCHWKernel::run(), CLGEMMLowpOffsetContributionKernel::run(), CLWinogradInputTransformKernel::run(), CLDepthwiseConvolutionLayer3x3NHWCKernel::run(), CLBatchNormalizationLayerKernel::run(), CLGEMMMatrixMultiplyKernel::run(), CLGEMMLowpMatrixMultiplyReshapedKernel::run(), CLDirectConvolutionLayerKernel::run(), CLColorConvertKernel::run(), CLGEMMMatrixMultiplyNativeKernel::run(), CLHOGBlockNormalizationKernel::run(), CLEdgeNonMaxSuppressionKernel::run(), CLWinogradOutputTransformKernel::run(), CLMinMaxLocationKernel::run(), CLGEMMLowpOffsetContributionOutputStageKernel::run(), CLIm2ColKernel::run(), CLCopyToArrayKernel::run(), CLSobel7x7VertKernel::run(), CLSobel5x5VertKernel::run(), CLGEMMReshapeRHSMatrixKernel::run(), CLGEMMLowpMatrixMultiplyReshapedOnlyRHSKernel::run(), CLLogits1DNormKernel::run(), CLGEMMMatrixMultiplyReshapedOnlyRHSKernel::run(), CLGEMMMatrixMultiplyReshapedKernel::run(), CLEdgeTraceKernel::run(), CLConvolutionRectangleKernel::run(), ClElementwiseKernel::run_op(), ClFloorKernel::run_op(), ClReshapeKernel::run_op(), ClCopyKernel::run_op(), ClElementWiseUnaryKernel::run_op(), ClWidthConcatenate2TensorsKernel::run_op(), ClActivationKernel::run_op(), ClPoolingKernel::run_op(), ClFillKernel::run_op(), ClDepthConcatenateKernel::run_op(), ClBatchConcatenateKernel::run_op(), ClWidthConcatenate4TensorsKernel::run_op(), CLStridedSliceKernel::run_op(), CLFillBorderKernel::run_op(), CLPixelWiseMultiplicationKernel::run_op(), CLComplexPixelWiseMultiplicationKernel::run_op(), Window::slide_window_slice_1D(), Window::slide_window_slice_2D(), Window::slide_window_slice_3D(), Window::slide_window_slice_4D(), Window::split_window(), and NEFFTDigitReverseKernel::validate().
template SimpleTensor<float> arm_compute::test::validation::reference::slice | ( | const SimpleTensor< float > & | src, |
Coordinates | starts, | ||
Coordinates | ends | ||
) |
template SimpleTensor<half_float::half> arm_compute::test::validation::reference::slice | ( | const SimpleTensor< half_float::half > & | src, |
Coordinates | starts, | ||
Coordinates | ends | ||
) |
std::pair< SimpleTensor< T >, SimpleTensor< T > > sobel | ( | const SimpleTensor< U > & | src, |
int | filter_size, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
GradientDimension | gradient_dimension | ||
) |
Definition at line 106 of file Sobel.cpp.
References SimpleTensor< T >::shape().
template std::pair< SimpleTensor< int >, SimpleTensor< int > > sobel | ( | const SimpleTensor< uint8_t > & | src, |
int | filter_size, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value, | ||
GradientDimension | gradient_dimension | ||
) |
SimpleTensor< T > softmax_layer | ( | const SimpleTensor< T > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::softmax_layer | ( | const SimpleTensor< float > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::softmax_layer | ( | const SimpleTensor< half > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::softmax_layer | ( | const SimpleTensor< uint8_t > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::softmax_layer | ( | const SimpleTensor< int8_t > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
SimpleTensor< T > softmax_layer_generic | ( | const SimpleTensor< T > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
Definition at line 39 of file SoftmaxLayer.cpp.
References ARM_COMPUTE_ERROR_ON, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::execute_window_loop(), offset(), Dimensions< T >::set(), Window::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, sum(), Window::use_tensor_dimensions(), and arm_compute::wrap_around().
template SimpleTensor<float> arm_compute::test::validation::reference::softmax_layer_generic | ( | const SimpleTensor< float > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::softmax_layer_generic | ( | const SimpleTensor< half > & | src, |
float | beta, | ||
int32_t | axis, | ||
bool | is_log | ||
) |
SimpleTensor< T > space_to_batch | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< int32_t > & | block_shape, | ||
const SimpleTensor< int32_t > & | paddings, | ||
const TensorShape & | dst_shape | ||
) |
Definition at line 38 of file SpaceToBatch.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst_shape, arm_compute::is_data_type_quantized(), SimpleTensor< T >::quantization_info(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::space_to_batch | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< int32_t > & | block_shape, | ||
const SimpleTensor< int32_t > & | paddings, | ||
const TensorShape & | dst_shape | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::space_to_batch | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< int32_t > & | block_shape, | ||
const SimpleTensor< int32_t > & | paddings, | ||
const TensorShape & | dst_shape | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::space_to_batch | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< int32_t > & | block_shape, | ||
const SimpleTensor< int32_t > & | paddings, | ||
const TensorShape & | dst_shape | ||
) |
SimpleTensor< T > space_to_depth | ( | const SimpleTensor< T > & | src, |
const TensorShape & | dst_shape, | ||
const int | block_shape | ||
) |
Definition at line 38 of file SpaceToDepth.cpp.
References SimpleTensor< T >::data_type(), arm_compute::test::validation::dst_shape, SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<float> arm_compute::test::validation::reference::space_to_depth | ( | const SimpleTensor< float > & | src, |
const TensorShape & | dst_shape, | ||
const int | block_shape | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::space_to_depth | ( | const SimpleTensor< half > & | src, |
const TensorShape & | dst_shape, | ||
const int | block_shape | ||
) |
SimpleTensor< T > stack_layer | ( | const std::vector< SimpleTensor< T >> & | in, |
const TensorShape & | output_shape, | ||
DataType | data_type, | ||
unsigned int | axis | ||
) |
Definition at line 41 of file StackLayer.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::data_type, Dimensions< T >::num_dimensions(), arm_compute::test::validation::output_shape, and arm_compute::test::validation::shape.
template SimpleTensor<int> arm_compute::test::validation::reference::stack_layer | ( | const std::vector< SimpleTensor< int >> & | in, |
const TensorShape & | output_shape, | ||
DataType | data_type, | ||
unsigned int | axis | ||
) |
template SimpleTensor<short> arm_compute::test::validation::reference::stack_layer | ( | const std::vector< SimpleTensor< short >> & | in, |
const TensorShape & | output_shape, | ||
DataType | data_type, | ||
unsigned int | axis | ||
) |
template SimpleTensor<char> arm_compute::test::validation::reference::stack_layer | ( | const std::vector< SimpleTensor< char >> & | in, |
const TensorShape & | output_shape, | ||
DataType | data_type, | ||
unsigned int | axis | ||
) |
SimpleTensor< T > strided_slice | ( | const SimpleTensor< T > & | src, |
Coordinates | starts, | ||
Coordinates | ends, | ||
BiStrides | strides, | ||
int32_t | begin_mask, | ||
int32_t | end_mask, | ||
int32_t | shrink_axis_mask | ||
) |
Definition at line 80 of file SliceOperations.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::helpers::tensor_transform::calculate_strided_slice_coords(), Dimensions< T >::cbegin(), arm_compute::helpers::tensor_transform::compute_strided_slice_output_shape(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, arm_compute::execute_window_loop(), Dimensions< T >::num_dimensions(), offset(), Dimensions< T >::set(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, and Window::use_tensor_dimensions().
template SimpleTensor<float> arm_compute::test::validation::reference::strided_slice | ( | const SimpleTensor< float > & | src, |
Coordinates | starts, | ||
Coordinates | ends, | ||
BiStrides | strides, | ||
int32_t | begin_mask, | ||
int32_t | end_mask, | ||
int32_t | shrink_axis_mask | ||
) |
template SimpleTensor<half_float::half> arm_compute::test::validation::reference::strided_slice | ( | const SimpleTensor< half_float::half > & | src, |
Coordinates | starts, | ||
Coordinates | ends, | ||
BiStrides | strides, | ||
int32_t | begin_mask, | ||
int32_t | end_mask, | ||
int32_t | shrink_axis_mask | ||
) |
SimpleTensor< T > table_lookup | ( | const SimpleTensor< T > & | src, |
const std::map< T, T > & | rawlut | ||
) |
Definition at line 37 of file TableLookup.cpp.
References SimpleTensor< T >::data_type(), SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), and arm_compute::test::validation::src.
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::table_lookup | ( | const SimpleTensor< uint8_t > & | src, |
const std::map< uint8_t, uint8_t > & | rawlut | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::table_lookup | ( | const SimpleTensor< int16_t > & | src, |
const std::map< int16_t, int16_t > & | rawlut | ||
) |
SimpleTensor< T > threshold | ( | const SimpleTensor< T > & | src, |
T | threshold, | ||
T | false_value, | ||
T | true_value, | ||
ThresholdType | type, | ||
T | upper | ||
) |
Definition at line 35 of file Threshold.cpp.
References ARM_COMPUTE_ERROR, arm_compute::BINARY, SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, SimpleTensor< T >::num_elements(), arm_compute::RANGE, SimpleTensor< T >::shape(), arm_compute::test::validation::src, and type.
Referenced by arm_compute::test::validation::asymm_rounding_divide_by_pow2(), compute_strength(), NEFastCornersKernel::configure(), NEFillArrayKernel::configure(), NEHOGDetectorKernel::configure(), CLFastCornersKernel::configure(), NEHOGMultiDetection::configure(), CLHOGMultiDetection::configure(), fast_corners(), fast_corners(), harris_corner_detector(), hog_detector(), hog_multi_detection(), NEHOGDetectorKernel::name(), NEFastCornersKernel::name(), NEFillArrayKernel::name(), arm_compute::quantization::rounding_divide_by_pow2(), arm_compute::rounding_divide_by_pow2(), arm_compute::quantization::saturating_rounding_multiply_by_pow2(), and NEThresholdKernel::validate().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::threshold | ( | const SimpleTensor< uint8_t > & | src, |
uint8_t | threshold, | ||
uint8_t | false_value, | ||
uint8_t | true_value, | ||
ThresholdType | type, | ||
uint8_t | upper | ||
) |
SimpleTensor< T > tile | ( | const SimpleTensor< T > & | src, |
const Multiples & | multiples | ||
) |
Definition at line 38 of file Tile.cpp.
References arm_compute::misc::shape_calculator::compute_tiled_shape(), arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::test::validation::w, Dimensions< T >::x(), Dimensions< T >::y(), and Dimensions< T >::z().
Referenced by arm_compute::test::validation::fill_lookuptable().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< uint8_t > & | src, |
const Multiples & | multiples | ||
) |
template SimpleTensor<int8_t> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< int8_t > & | src, |
const Multiples & | multiples | ||
) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< uint16_t > & | src, |
const Multiples & | multiples | ||
) |
template SimpleTensor<int16_t> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< int16_t > & | src, |
const Multiples & | multiples | ||
) |
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< uint32_t > & | src, |
const Multiples & | multiples | ||
) |
template SimpleTensor<int32_t> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< int32_t > & | src, |
const Multiples & | multiples | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< half > & | src, |
const Multiples & | multiples | ||
) |
template SimpleTensor<float> arm_compute::test::validation::reference::tile | ( | const SimpleTensor< float > & | src, |
const Multiples & | multiples | ||
) |
SimpleTensor< T > transpose | ( | const SimpleTensor< T > & | src | ) |
Definition at line 38 of file Transpose.cpp.
References arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), SimpleTensor< T >::shape(), arm_compute::test::validation::src, Dimensions< T >::x(), and Dimensions< T >::y().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::transpose | ( | const SimpleTensor< uint8_t > & | src | ) |
template SimpleTensor<uint16_t> arm_compute::test::validation::reference::transpose | ( | const SimpleTensor< uint16_t > & | src | ) |
template SimpleTensor<uint32_t> arm_compute::test::validation::reference::transpose | ( | const SimpleTensor< uint32_t > & | src | ) |
template SimpleTensor<half> arm_compute::test::validation::reference::transpose | ( | const SimpleTensor< half > & | src | ) |
template SimpleTensor<float> arm_compute::test::validation::reference::transpose | ( | const SimpleTensor< float > & | src | ) |
std::vector< SimpleTensor< T > > unstack | ( | const SimpleTensor< T > & | input_tensor, |
std::vector< SimpleTensor< T >> & | output_tensors, | ||
int | axis | ||
) |
Definition at line 87 of file Unstack.cpp.
References ARM_COMPUTE_ERROR_ON, SimpleTensor< T >::shape(), and arm_compute::wrap_around().
template std::vector<SimpleTensor<float> > arm_compute::test::validation::reference::unstack | ( | const SimpleTensor< float > & | input_tensor, |
std::vector< SimpleTensor< float >> & | output_tensors, | ||
int | axis | ||
) |
template std::vector<SimpleTensor<half> > arm_compute::test::validation::reference::unstack | ( | const SimpleTensor< half > & | input_tensor, |
std::vector< SimpleTensor< half >> & | output_tensors, | ||
int | axis | ||
) |
template std::vector<SimpleTensor<uint8_t> > arm_compute::test::validation::reference::unstack | ( | const SimpleTensor< uint8_t > & | input_tensor, |
std::vector< SimpleTensor< uint8_t >> & | output_tensors, | ||
int | axis | ||
) |
bool arm_compute::test::validation::reference::valid_bilinear_policy | ( | float | xn, |
float | yn, | ||
int | width, | ||
int | height, | ||
BorderMode | border_mode | ||
) |
Definition at line 36 of file WarpAffine.cpp.
References arm_compute::UNDEFINED.
Referenced by remap(), warp_affine(), and warp_perspective().
SimpleTensor< T > warp_affine | ( | const SimpleTensor< T > & | src, |
SimpleTensor< T > & | valid_mask, | ||
const float * | matrix, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
Definition at line 50 of file WarpAffine.cpp.
References arm_compute::AREA, ARM_COMPUTE_ERROR, arm_compute::BILINEAR, arm_compute::test::validation::bilinear_policy(), arm_compute::CONSTANT, arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), arm_compute::NEAREST_NEIGHBOR, SimpleTensor< T >::num_elements(), arm_compute::REPLICATE, SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), arm_compute::UNDEFINED, valid_bilinear_policy(), and Dimensions< T >::x().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::warp_affine | ( | const SimpleTensor< uint8_t > & | src, |
SimpleTensor< uint8_t > & | valid_mask, | ||
const float * | matrix, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > warp_perspective | ( | const SimpleTensor< T > & | src, |
SimpleTensor< T > & | valid_mask, | ||
const float * | matrix, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
Definition at line 38 of file WarpPerspective.cpp.
References arm_compute::AREA, ARM_COMPUTE_ERROR, arm_compute::BILINEAR, arm_compute::test::validation::bilinear_policy(), arm_compute::CONSTANT, arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), arm_compute::NEAREST_NEIGHBOR, SimpleTensor< T >::num_elements(), arm_compute::REPLICATE, SimpleTensor< T >::shape(), arm_compute::test::validation::src, arm_compute::test::validation::tensor_elem_at(), arm_compute::UNDEFINED, valid_bilinear_policy(), and Dimensions< T >::x().
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::warp_perspective | ( | const SimpleTensor< uint8_t > & | src, |
SimpleTensor< uint8_t > & | valid_mask, | ||
const float * | matrix, | ||
InterpolationPolicy | policy, | ||
BorderMode | border_mode, | ||
uint8_t | constant_border_value | ||
) |
SimpleTensor< T > weights_reshape | ( | const SimpleTensor< T > & | src, |
const SimpleTensor< T > & | biases, | ||
const TensorShape & | dst_shape, | ||
const unsigned int | num_groups | ||
) |
Definition at line 38 of file WeightsReshape.cpp.
References arm_compute::test::coord2index(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::validation::dst_shape, arm_compute::test::validation::has_bias, arm_compute::test::validation::num_groups, SimpleTensor< T >::shape(), SimpleTensor< T >::size(), arm_compute::test::validation::src, and arm_compute::test::validation::w.
template SimpleTensor<float> arm_compute::test::validation::reference::weights_reshape | ( | const SimpleTensor< float > & | src, |
const SimpleTensor< float > & | biases, | ||
const TensorShape & | dst_shape, | ||
const unsigned int | num_groups | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::weights_reshape | ( | const SimpleTensor< half > & | src, |
const SimpleTensor< half > & | biases, | ||
const TensorShape & | dst_shape, | ||
const unsigned int | num_groups | ||
) |
template SimpleTensor<uint8_t> arm_compute::test::validation::reference::weights_reshape | ( | const SimpleTensor< uint8_t > & | src, |
const SimpleTensor< uint8_t > & | biases, | ||
const TensorShape & | dst_shape, | ||
const unsigned int | num_groups | ||
) |
SimpleTensor< T > winograd_filter_transform | ( | const SimpleTensor< T > & | in, |
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
Definition at line 360 of file Winograd.cpp.
References Size2D::area(), ARM_COMPUTE_ERROR_ON_MSG, SimpleTensor< T >::data_layout(), SimpleTensor< T >::data_type(), FILTER, Size2D::height, WinogradInfo::kernel_size, arm_compute::NCHW, arm_compute::test::validation::output_shape, WinogradInfo::output_tile_size, SimpleTensor< T >::shape(), arm_compute::test::validation::w, and Size2D::width.
Referenced by winograd_output_transform().
template SimpleTensor<float> arm_compute::test::validation::reference::winograd_filter_transform | ( | const SimpleTensor< float > & | in, |
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::winograd_filter_transform | ( | const SimpleTensor< half > & | in, |
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
SimpleTensor< T > winograd_input_transform | ( | const SimpleTensor< T > & | in, |
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
Definition at line 236 of file Winograd.cpp.
References ARM_COMPUTE_ERROR_ON, arm_compute::test::validation::b, arm_compute::compute_winograd_convolution_tiles(), arm_compute::test::validation::conv_info, WinogradInfo::convolution_info, arm_compute::coords2index(), SimpleTensor< T >::data_layout(), SimpleTensor< T >::data_type(), Size2D::height, INPUT, WinogradInfo::kernel_size, arm_compute::NCHW, arm_compute::test::validation::output_shape, WinogradInfo::output_tile_size, SimpleTensor< T >::shape(), and Size2D::width.
Referenced by winograd_output_transform().
template SimpleTensor<float> arm_compute::test::validation::reference::winograd_input_transform | ( | const SimpleTensor< float > & | in, |
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::winograd_input_transform | ( | const SimpleTensor< half > & | in, |
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
SimpleTensor< T > winograd_output_transform | ( | const SimpleTensor< T > & | in, |
const SimpleTensor< T > & | b, | ||
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
Definition at line 440 of file Winograd.cpp.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, arm_compute::test::validation::b, arm_compute::CHANNEL, arm_compute::compute_winograd_convolution_tiles(), arm_compute::test::validation::conv_info, WinogradInfo::convolution_info, SimpleTensor< T >::data_type(), arm_compute::get_data_layout_dimension_index(), Size2D::height, WinogradInfo::input_dimensions, WinogradInfo::kernel_size, OUTPUT, WinogradInfo::output_data_layout, arm_compute::test::validation::output_shape, WinogradInfo::output_tile_size, SimpleTensor< T >::shape(), Size2D::width, winograd_filter_transform(), arm_compute::test::validation::winograd_info, and winograd_input_transform().
template SimpleTensor<float> arm_compute::test::validation::reference::winograd_output_transform | ( | const SimpleTensor< float > & | in, |
const SimpleTensor< float > & | b, | ||
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::winograd_output_transform | ( | const SimpleTensor< half > & | in, |
const SimpleTensor< half > & | b, | ||
const TensorShape & | output_shape, | ||
const WinogradInfo & | winograd_info | ||
) |
SimpleTensor< T > yolo_layer | ( | const SimpleTensor< T > & | src, |
const ActivationLayerInfo & | info, | ||
int32_t | num_classes | ||
) |
Definition at line 40 of file YOLOLayer.cpp.
References ActivationLayerInfo::a(), ActivationLayerInfo::activation(), arm_compute::test::validation::b, ActivationLayerInfo::b(), SimpleTensor< T >::data_type(), arm_compute::test::validation::dst, arm_compute::test::index2coord(), SimpleTensor< T >::num_elements(), and SimpleTensor< T >::shape().
Referenced by yolo_layer< uint8_t >().
template SimpleTensor<float> arm_compute::test::validation::reference::yolo_layer | ( | const SimpleTensor< float > & | src, |
const ActivationLayerInfo & | info, | ||
int32_t | num_classes | ||
) |
template SimpleTensor<half> arm_compute::test::validation::reference::yolo_layer | ( | const SimpleTensor< half > & | src, |
const ActivationLayerInfo & | info, | ||
int32_t | num_classes | ||
) |
SimpleTensor<uint8_t> arm_compute::test::validation::reference::yolo_layer< uint8_t > | ( | const SimpleTensor< uint8_t > & | src, |
const ActivationLayerInfo & | info, | ||
int32_t | num_classes | ||
) |
Definition at line 71 of file YOLOLayer.cpp.
References arm_compute::test::validation::convert_from_asymmetric(), arm_compute::test::validation::dst, arm_compute::test::validation::info, SimpleTensor< T >::quantization_info(), and yolo_layer().