| /external/ComputeLibrary/tests/validation/reference/ |
| D | DepthConvertLayer.cpp | 42 SimpleTensor<T2> depth_convert(const SimpleTensor<T1> &src, DataType dt_out, ConvertPolicy policy, … in depth_convert()
44 SimpleTensor<T2> result(src.shape(), dt_out); in depth_convert()
73 SimpleTensor<T2> depth_convert(const SimpleTensor<T1> &src, DataType dt_out, ConvertPolicy policy, … in depth_convert()
75 SimpleTensor<T2> result(src.shape(), dt_out); in depth_convert()
99 SimpleTensor<T2> depth_convert(const SimpleTensor<T1> &src, DataType dt_out, ConvertPolicy policy, … in depth_convert()
101 SimpleTensor<T2> result(src.shape(), dt_out); in depth_convert()
114 template SimpleTensor<int8_t> depth_convert(const SimpleTensor<uint8_t> &src, DataType dt_out, Conv…
115 template SimpleTensor<uint16_t> depth_convert(const SimpleTensor<uint8_t> &src, DataType dt_out, Co…
116 template SimpleTensor<int16_t> depth_convert(const SimpleTensor<uint8_t> &src, DataType dt_out, Con…
117 template SimpleTensor<uint32_t> depth_convert(const SimpleTensor<uint8_t> &src, DataType dt_out, Co…
[all …]
|
| D | FuseBatchNormalization.cpp | 36 …st SimpleTensor<T> &w, const SimpleTensor<T> &mean, const SimpleTensor<T> &var, SimpleTensor<T> &w… in fuse_batch_normalization_dwc_layer()
37 … const SimpleTensor<T> &beta, const SimpleTensor<T> &gamma, float epsilon) in fuse_batch_normalization_dwc_layer()
71 …n_conv_layer(const SimpleTensor<T> &w, const SimpleTensor<T> &mean, const SimpleTensor<T> &var, Si… in fuse_batch_normalization_conv_layer()
72 const SimpleTensor<T> &b, in fuse_batch_normalization_conv_layer()
73 … const SimpleTensor<T> &beta, const SimpleTensor<T> &gamma, float epsilon) in fuse_batch_normalization_conv_layer()
104 …alization_dwc_layer(const SimpleTensor<float> &w, const SimpleTensor<float> &mean, const SimpleTen…
105 …SimpleTensor<float> &b_fused, const SimpleTensor<float> &b, const SimpleTensor<float> &beta, const…
106 …_layer(const SimpleTensor<half> &w, const SimpleTensor<half> &mean, const SimpleTensor<half> &var,…
107 …const SimpleTensor<half> &b, const SimpleTensor<half> &beta, const SimpleTensor<half> &gamma, floa…
108 …lization_conv_layer(const SimpleTensor<float> &w, const SimpleTensor<float> &mean, const SimpleTen…
[all …]
|
| D | ArithmeticOperations.cpp | 56 …c void unroll(ArithmeticOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
83 …c void unroll(ArithmeticOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
92 SimpleTensor<T> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<T> &src1, const Sim… in arithmetic_operation()
104 SimpleTensor<uint8_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<uint8_t> &src… in arithmetic_operation()
112 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in arithmetic_operation()
113 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in arithmetic_operation()
114 …SimpleTensor<float> dst_tmp(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dst.data_typ… in arithmetic_operation()
131 SimpleTensor<int8_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int8_t> &src1,… in arithmetic_operation()
139 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in arithmetic_operation()
140 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in arithmetic_operation()
[all …]
|
| D | NormalizePlanarYUVLayer.cpp | 38 SimpleTensor<T> normalize_planar_yuv_layer(const SimpleTensor<T> &src, const SimpleTensor<T> &mean,… in normalize_planar_yuv_layer()
40 SimpleTensor<T> result(src.shape(), src.data_type()); in normalize_planar_yuv_layer()
65 SimpleTensor<uint8_t> normalize_planar_yuv_layer<uint8_t>(const SimpleTensor<uint8_t> &src, const S… in normalize_planar_yuv_layer()
67 SimpleTensor<float> src_tmp = convert_from_asymmetric(src); in normalize_planar_yuv_layer()
68 SimpleTensor<float> mean_tmp = convert_from_asymmetric(mean); in normalize_planar_yuv_layer()
69 SimpleTensor<float> std_tmp = convert_from_asymmetric(std); in normalize_planar_yuv_layer()
70 SimpleTensor<float> dst_tmp = normalize_planar_yuv_layer<float>(src_tmp, mean_tmp, std_tmp); in normalize_planar_yuv_layer()
71 … SimpleTensor<uint8_t> dst = convert_to_asymmetric<uint8_t>(dst_tmp, src.quantization_info()); in normalize_planar_yuv_layer()
76 SimpleTensor<int8_t> normalize_planar_yuv_layer<int8_t>(const SimpleTensor<int8_t> &src, const Simp… in normalize_planar_yuv_layer()
78 SimpleTensor<float> src_tmp = convert_from_asymmetric(src); in normalize_planar_yuv_layer()
[all …]
|
| D | ElementwiseOperations.cpp | 121 …c void unroll(ArithmeticOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
145 …c void unroll(ArithmeticOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
154 SimpleTensor<T> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<T> &src1, const Sim… in arithmetic_operation()
166 SimpleTensor<uint8_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<uint8_t> &src… in arithmetic_operation()
170 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in arithmetic_operation()
171 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in arithmetic_operation()
172 …SimpleTensor<float> dst_tmp(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dst.data_typ… in arithmetic_operation()
196 SimpleTensor<int8_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int8_t> &src1,… in arithmetic_operation()
200 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in arithmetic_operation()
201 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in arithmetic_operation()
[all …]
|
| D | GEMMLowp.cpp | 66 void quantize_down_scale(const SimpleTensor<TIn> *in, const SimpleTensor<TIn> *bias, SimpleTensor<T… in quantize_down_scale()
100 …id quantize_down_scale_by_fixedpoint(const SimpleTensor<TIn> *in, const SimpleTensor<TIn> *bias, S… in quantize_down_scale_by_fixedpoint()
144 void quantize_down_scale_by_float(const SimpleTensor<TIn> *in, const SimpleTensor<TIn> *bias, Simpl… in quantize_down_scale_by_float()
181 SimpleTensor<T_out> gemmlowp_matrix_multiply_core(const SimpleTensor<T_in> &a, const SimpleTensor<T… in gemmlowp_matrix_multiply_core()
186 SimpleTensor<T_out> c(shape_c, dt); in gemmlowp_matrix_multiply_core()
235 SimpleTensor<T1> gemmlowp(const SimpleTensor<T2> &a, const SimpleTensor<T3> &b, TensorShape shape_c) in gemmlowp()
241 SimpleTensor<TOut> gemmlowp_quantize_down_scale(const SimpleTensor<TIn> &in, int32_t result_offset,… in gemmlowp_quantize_down_scale()
244 SimpleTensor<TOut> dst(in.shape(), DataTypeExtractor<TOut>::data_type()); in gemmlowp_quantize_down_scale()
252 SimpleTensor<TOut> gemmlowp_quantize_down_scale(const SimpleTensor<TIn> &in, const SimpleTensor<TIn… in gemmlowp_quantize_down_scale()
255 SimpleTensor<TOut> dst(in.shape(), DataTypeExtractor<TOut>::data_type()); in gemmlowp_quantize_down_scale()
[all …]
|
| D | QuantizationLayer.cpp | 37 SimpleTensor<Tout> quantization_layer(const SimpleTensor<Tin> &src, DataType output_data_type, cons… in quantization_layer()
40 SimpleTensor<Tout> dst{ src.shape(), output_data_type, 1, quantization_info }; in quantization_layer()
90 SimpleTensor<uint8_t> quantization_layer(const SimpleTensor<uint8_t> &src, DataType output_data_typ… in quantization_layer()
92 SimpleTensor<float> src_tmp = convert_from_asymmetric<uint8_t>(src); in quantization_layer()
97 SimpleTensor<int8_t> quantization_layer(const SimpleTensor<uint8_t> &src, DataType output_data_type… in quantization_layer()
99 SimpleTensor<float> src_tmp = convert_from_asymmetric<uint8_t>(src); in quantization_layer()
104 SimpleTensor<uint8_t> quantization_layer(const SimpleTensor<int8_t> &src, DataType output_data_type… in quantization_layer()
106 SimpleTensor<float> src_tmp = convert_from_asymmetric<int8_t>(src); in quantization_layer()
111 SimpleTensor<int8_t> quantization_layer(const SimpleTensor<int8_t> &src, DataType output_data_type,… in quantization_layer()
113 SimpleTensor<float> src_tmp = convert_from_asymmetric<int8_t>(src); in quantization_layer()
[all …]
|
| D | Copy.cpp | 37 SimpleTensor<T> copy(const SimpleTensor<T> &src, const TensorShape &output_shape) in copy()
41 SimpleTensor<T> dst(output_shape, src.data_type()); in copy()
46 template SimpleTensor<uint8_t> copy(const SimpleTensor<uint8_t> &src, const TensorShape &output_sha…
47 template SimpleTensor<int8_t> copy(const SimpleTensor<int8_t> &src, const TensorShape &output_shape…
48 template SimpleTensor<uint16_t> copy(const SimpleTensor<uint16_t> &src, const TensorShape &output_s…
49 template SimpleTensor<int16_t> copy(const SimpleTensor<int16_t> &src, const TensorShape &output_sha…
50 template SimpleTensor<uint32_t> copy(const SimpleTensor<uint32_t> &src, const TensorShape &output_s…
51 template SimpleTensor<int32_t> copy(const SimpleTensor<int32_t> &src, const TensorShape &output_sha…
52 template SimpleTensor<half> copy(const SimpleTensor<half> &src, const TensorShape &output_shape);
53 template SimpleTensor<float> copy(const SimpleTensor<float> &src, const TensorShape &output_shape);
|
| D | ActivationLayer.cpp | 38 SimpleTensor<T> activation_layer(const SimpleTensor<T> &src, ActivationLayerInfo info, const Quanti… in activation_layer()
43 SimpleTensor<T> dst{ src.shape(), src.data_type(), 1 }; in activation_layer()
60 SimpleTensor<uint8_t> activation_layer<uint8_t>(const SimpleTensor<uint8_t> &src, ActivationLayerIn… in activation_layer()
64 SimpleTensor<float> src_tmp = convert_from_asymmetric(src); in activation_layer()
65 SimpleTensor<float> dst_tmp = activation_layer<float>(src_tmp, info); in activation_layer()
66 SimpleTensor<uint8_t> dst = convert_to_asymmetric<uint8_t>(dst_tmp, dst_qinfo); in activation_layer()
71 SimpleTensor<int8_t> activation_layer<int8_t>(const SimpleTensor<int8_t> &src, ActivationLayerInfo … in activation_layer()
75 SimpleTensor<float> src_tmp = convert_from_asymmetric(src); in activation_layer()
76 SimpleTensor<float> dst_tmp = activation_layer<float>(src_tmp, info); in activation_layer()
77 SimpleTensor<int8_t> dst = convert_to_asymmetric<int8_t>(dst_tmp, dst_qinfo); in activation_layer()
[all …]
|
| D | CropResize.cpp | 37 SimpleTensor<float> scale_image(const SimpleTensor<float> &in, const TensorShape &out_shape, Interp… in scale_image()
41 SimpleTensor<float> out{ out_shape, DataType::F32, 1, QuantizationInfo(), DataLayout::NHWC }; in scale_image()
124 SimpleTensor<float> crop_image(const SimpleTensor<T> &src, Coordinates start, Coordinates end, int3… in crop_image()
128 SimpleTensor<float> out{ out_shape, DataType::F32, 1, QuantizationInfo(), DataLayout::NHWC }; in crop_image()
160 SimpleTensor<float> crop_and_resize(const SimpleTensor<T> &src, const SimpleTensor<float> &boxes, S… in crop_and_resize()
167 SimpleTensor<float> out{ out_shape, DataType::F32, 1, QuantizationInfo(), DataLayout::NHWC }; in crop_and_resize()
177 SimpleTensor<float> cropped = crop_image(src, start, end, box_ind[i], extrapolation_value); in crop_and_resize()
178 …SimpleTensor<float> scaled = scale_image(cropped, scaled_image_shape, method, extrapolation_value… in crop_and_resize()
184 template SimpleTensor<float> crop_and_resize(const SimpleTensor<float> &src, const SimpleTensor<flo…
186 template SimpleTensor<float> crop_and_resize(const SimpleTensor<uint16_t> &src, const SimpleTensor<…
[all …]
|
| D | ReshapeLayer.cpp | 38 SimpleTensor<T> reshape_layer(const SimpleTensor<T> &src, const TensorShape &output_shape) in reshape_layer()
42 SimpleTensor<T> dst(output_shape, src.data_type()); in reshape_layer()
47 template SimpleTensor<uint8_t> reshape_layer(const SimpleTensor<uint8_t> &src, const TensorShape &o…
48 template SimpleTensor<int8_t> reshape_layer(const SimpleTensor<int8_t> &src, const TensorShape &out…
49 template SimpleTensor<uint16_t> reshape_layer(const SimpleTensor<uint16_t> &src, const TensorShape …
50 template SimpleTensor<int16_t> reshape_layer(const SimpleTensor<int16_t> &src, const TensorShape &o…
51 template SimpleTensor<uint32_t> reshape_layer(const SimpleTensor<uint32_t> &src, const TensorShape …
52 template SimpleTensor<int32_t> reshape_layer(const SimpleTensor<int32_t> &src, const TensorShape &o…
53 template SimpleTensor<half> reshape_layer(const SimpleTensor<half> &src, const TensorShape &output_…
54 template SimpleTensor<float> reshape_layer(const SimpleTensor<float> &src, const TensorShape &outpu…
|
| D | ConcatenateLayer.cpp | 40 SimpleTensor<T> widthconcatenate_layer(const std::vector<SimpleTensor<T>> &srcs, SimpleTensor<T> &d… in widthconcatenate_layer()
109 template SimpleTensor<float> widthconcatenate_layer(const std::vector<SimpleTensor<float>> &srcs, S…
110 template SimpleTensor<half> widthconcatenate_layer(const std::vector<SimpleTensor<half>> &srcs, Sim…
111 template SimpleTensor<uint8_t> widthconcatenate_layer(const std::vector<SimpleTensor<uint8_t>> &src…
112 template SimpleTensor<int8_t> widthconcatenate_layer(const std::vector<SimpleTensor<int8_t>> &srcs,…
116 SimpleTensor<T> concatenate_layer(std::vector<SimpleTensor<T>> &srcs, SimpleTensor<T> &dst, unsigne… in concatenate_layer()
160 template SimpleTensor<float> concatenate_layer(std::vector<SimpleTensor<float>> &srcs, SimpleTensor…
161 template SimpleTensor<half> concatenate_layer(std::vector<SimpleTensor<half>> &srcs, SimpleTensor<h…
162 template SimpleTensor<uint8_t> concatenate_layer(std::vector<SimpleTensor<uint8_t>> &srcs, SimpleTe…
163 template SimpleTensor<int8_t> concatenate_layer(std::vector<SimpleTensor<int8_t>> &srcs, SimpleTens…
|
| D | PixelWiseMultiplication.cpp | 182 …static void unroll(const SimpleTensor<T1> &src1, const SimpleTensor<T2> &src2, SimpleTensor<T3> &d… in unroll()
207 …static void unroll(const SimpleTensor<T1> &src1, const SimpleTensor<T2> &src2, SimpleTensor<T3> &d… in unroll()
217 SimpleTensor<T3> pixel_wise_multiplication(const SimpleTensor<T1> &src1, const SimpleTensor<T2> &sr… in pixel_wise_multiplication()
222 SimpleTensor<T3> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dt_out); in pixel_wise_multiplication()
239 SimpleTensor<uint8_t> pixel_wise_multiplication(const SimpleTensor<uint8_t> &src1, const SimpleTens… in pixel_wise_multiplication()
242 …SimpleTensor<uint8_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dt_out, 1, qou… in pixel_wise_multiplication()
246 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in pixel_wise_multiplication()
247 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in pixel_wise_multiplication()
248 …SimpleTensor<float> dst_tmp = pixel_wise_multiplication<float, float, float>(src1_tmp, src2_tmp, … in pixel_wise_multiplication()
267 SimpleTensor<int16_t> pixel_wise_multiplication(const SimpleTensor<uint8_t> &src1, const SimpleTens… in pixel_wise_multiplication()
[all …]
|
| D | ConvolutionLayer.cpp | 45 SimpleTensor<T> convolution_layer_nchw(const SimpleTensor<T> &src, const SimpleTensor<TW> &weights,… in convolution_layer_nchw()
111 SimpleTensor<T> convolution_layer(const SimpleTensor<T> &src, const SimpleTensor<TW> &weights, cons… in convolution_layer()
120 SimpleTensor<T> dst{ output_shape, src.data_type(), 1, out_quant_info }; in convolution_layer()
125 template SimpleTensor<float> convolution_layer(const SimpleTensor<float> &src, const SimpleTensor<f…
127 template SimpleTensor<half> convolution_layer(const SimpleTensor<half> &src, const SimpleTensor<hal…
129 template SimpleTensor<uint8_t> convolution_layer(const SimpleTensor<uint8_t> &src, const SimpleTens…
131 template SimpleTensor<uint8_t> convolution_layer(const SimpleTensor<uint8_t> &src, const SimpleTens…
133 template SimpleTensor<int8_t> convolution_layer(const SimpleTensor<int8_t> &src, const SimpleTensor…
|
| D | FuseBatchNormalization.h | 39 …st SimpleTensor<T> &w, const SimpleTensor<T> &mean, const SimpleTensor<T> &var, SimpleTensor<T> &w…
40 … const SimpleTensor<T> &beta, const SimpleTensor<T> &gamma, float epsilon);
43 …n_conv_layer(const SimpleTensor<T> &w, const SimpleTensor<T> &mean, const SimpleTensor<T> &var, Si…
44 const SimpleTensor<T> &b,
45 … const SimpleTensor<T> &beta, const SimpleTensor<T> &gamma, float epsilon);
|
| D | DFT.cpp | 179 SimpleTensor<T> rdft_1d_core(const SimpleTensor<T> &src, FFTDirection direction, bool is_odd) in rdft_1d_core()
193 SimpleTensor<T> dst(dst_shape, src.data_type(), num_channels); in rdft_1d_core()
210 SimpleTensor<T> dft_1d_core(const SimpleTensor<T> &src, FFTDirection direction) in dft_1d_core()
216 SimpleTensor<T> dst(src.shape(), src.data_type(), src.num_channels()); in dft_1d_core()
238 void scale(SimpleTensor<T> &tensor, T scaling_factor) in scale()
259 SimpleTensor<T> complex_mul_and_reduce(const SimpleTensor<T> &input, const SimpleTensor<T> &weights) in complex_mul_and_reduce()
269 SimpleTensor<T> dst(output_shape, input.data_type(), input.num_channels()); in complex_mul_and_reduce()
313 SimpleTensor<T> rdft_1d(const SimpleTensor<T> &src) in rdft_1d()
319 SimpleTensor<T> ridft_1d(const SimpleTensor<T> &src, bool is_odd) in ridft_1d()
330 SimpleTensor<T> dft_1d(const SimpleTensor<T> &src, FFTDirection direction) in dft_1d()
[all …]
|
| D | GEMMLowp.h | 39 SimpleTensor<T1> gemmlowp_matrix_multiply_core(const SimpleTensor<T2> &a, const SimpleTensor<T3> &b…
42 SimpleTensor<T1> gemmlowp(const SimpleTensor<T2> &a, const SimpleTensor<T3> &b, TensorShape shape_c…
45 SimpleTensor<uint8_t> gemmlowp_quantize_down_scale(const SimpleTensor<TIn> &in, int32_t result_offs…
48 SimpleTensor<TOut> gemmlowp_quantize_down_scale(const SimpleTensor<TIn> &in, int32_t result_offset,…
52 SimpleTensor<TOut> gemmlowp_quantize_down_scale(const SimpleTensor<TIn> &in, const SimpleTensor<TIn…
56 SimpleTensor<TOut> gemmlowp_quantize_down_scale_by_fixedpoint(const SimpleTensor<TIn> &in, std::vec…
60 SimpleTensor<TOut> gemmlowp_quantize_down_scale_by_fixedpoint(const SimpleTensor<TIn> &in, const Si…
64 SimpleTensor<TOut> gemmlowp_quantize_down_scale_by_float(const SimpleTensor<TIn> &in, const SimpleT…
68 SimpleTensor<TOut> gemmlowp_quantize_down_scale_by_float(const SimpleTensor<TIn> &in,
|
| D | Tile.cpp | 38 SimpleTensor<T> tile(const SimpleTensor<T> &src, const Multiples &multiples) in tile()
44 SimpleTensor<T> dst{ tiled_shape, src.data_type() }; in tile()
65 template SimpleTensor<uint8_t> tile(const SimpleTensor<uint8_t> &src, const Multiples &multiples);
66 template SimpleTensor<int8_t> tile(const SimpleTensor<int8_t> &src, const Multiples &multiples);
67 template SimpleTensor<uint16_t> tile(const SimpleTensor<uint16_t> &src, const Multiples &multiples);
68 template SimpleTensor<int16_t> tile(const SimpleTensor<int16_t> &src, const Multiples &multiples);
69 template SimpleTensor<uint32_t> tile(const SimpleTensor<uint32_t> &src, const Multiples &multiples);
70 template SimpleTensor<int32_t> tile(const SimpleTensor<int32_t> &src, const Multiples &multiples);
71 template SimpleTensor<half> tile(const SimpleTensor<half> &src, const Multiples &multiples);
72 template SimpleTensor<float> tile(const SimpleTensor<float> &src, const Multiples &multiples);
|
| D | Logical.cpp | 60 …oll(arm_compute::LogicalOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
87 …oll(arm_compute::LogicalOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
95 SimpleTensor<T> logical_or(const SimpleTensor<T> &src1, const SimpleTensor<T> &src2) in logical_or()
100 … SimpleTensor<T> dst{ TensorShape::broadcast_shape(src1.shape(), src2.shape()), src1.data_type() }; in logical_or()
108 SimpleTensor<T> logical_and(const SimpleTensor<T> &src1, const SimpleTensor<T> &src2) in logical_and()
113 … SimpleTensor<T> dst{ TensorShape::broadcast_shape(src1.shape(), src2.shape()), src1.data_type() }; in logical_and()
121 SimpleTensor<T> logical_not(const SimpleTensor<T> &src) in logical_not()
123 SimpleTensor<T> dst(src.shape(), src.data_type()); in logical_not()
135 template SimpleTensor<uint8_t> logical_or(const SimpleTensor<uint8_t> &src1, const SimpleTensor<uin…
136 template SimpleTensor<uint8_t> logical_and(const SimpleTensor<uint8_t> &src1, const SimpleTensor<ui…
[all …]
|
| D | BatchNormalizationLayer.cpp | 40 SimpleTensor<T> batch_normalization_layer(const SimpleTensor<T> &src, const SimpleTensor<T> &mean, … in batch_normalization_layer()
43 SimpleTensor<T> result(src.shape(), src.data_type()); in batch_normalization_layer()
77 …SimpleTensor<float> batch_normalization_layer(const SimpleTensor<float> &src, const SimpleTensor<f…
78 … const SimpleTensor<float> &gamma, float epsilon, ActivationLayerInfo act_info);
79 template SimpleTensor<half> batch_normalization_layer(const SimpleTensor<half> &src, const SimpleTe…
80 const SimpleTensor<half> &beta,
81 … const SimpleTensor<half> &gamma, float epsilon, ActivationLayerInfo act_info);
|
| D | Comparisons.cpp | 74 … const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, SimpleTensor<uint8_t> &dst, in unroll()
102 … const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, SimpleTensor<uint8_t> &dst, in unroll()
111 SimpleTensor<uint8_t> compare(ComparisonOperation op, const SimpleTensor<T> &src1, const SimpleTens… in compare()
113 … SimpleTensor<uint8_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), DataType::U8); in compare()
123 SimpleTensor<uint8_t> compare(ComparisonOperation op, const SimpleTensor<uint8_t> &src1, const Simp… in compare()
125 … SimpleTensor<uint8_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), DataType::U8); in compare()
133 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in compare()
134 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in compare()
146 SimpleTensor<uint8_t> compare(ComparisonOperation op, const SimpleTensor<int8_t> &src1, const Simpl… in compare()
148 … SimpleTensor<uint8_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), DataType::U8); in compare()
[all …]
|
| D | FullyConnectedLayer.cpp | 45 …ector_matrix_multiply(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, const SimpleTens… in vector_matrix_multiply()
63 …ector_matrix_multiply(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, const SimpleTens… in vector_matrix_multiply()
115 SimpleTensor<T> fully_connected_layer(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, c… in fully_connected_layer()
124 SimpleTensor<T> dst{ TensorShape{ dst_shape }, src.data_type(), 1, out_quant_info }; in fully_connected_layer()
161 template SimpleTensor<float> fully_connected_layer(const SimpleTensor<float> &src, const SimpleTens…
163 template SimpleTensor<half> fully_connected_layer(const SimpleTensor<half> &src, const SimpleTensor…
165 template SimpleTensor<uint8_t> fully_connected_layer(const SimpleTensor<uint8_t> &src, const Simple…
167 template SimpleTensor<int8_t> fully_connected_layer(const SimpleTensor<int8_t> &src, const SimpleTe…
|
| D | Permute.cpp | 38 SimpleTensor<T> permute(const SimpleTensor<T> &src, PermutationVector perm) in permute()
45 SimpleTensor<T> dst{ dst_shape, src.data_type(), src.num_channels(), src.quantization_info() }; in permute()
61 template SimpleTensor<int8_t> permute(const SimpleTensor<int8_t> &src, PermutationVector perm);
62 template SimpleTensor<uint8_t> permute(const SimpleTensor<uint8_t> &src, PermutationVector perm);
63 template SimpleTensor<int16_t> permute(const SimpleTensor<int16_t> &src, PermutationVector perm);
64 template SimpleTensor<uint16_t> permute(const SimpleTensor<uint16_t> &src, PermutationVector perm);
65 template SimpleTensor<uint32_t> permute(const SimpleTensor<uint32_t> &src, PermutationVector perm);
66 template SimpleTensor<float> permute(const SimpleTensor<float> &src, PermutationVector perm);
67 template SimpleTensor<half> permute(const SimpleTensor<half> &src, PermutationVector perm);
|
| /external/ComputeLibrary/tests/validation/ |
| D | Helpers.cpp | 36 SimpleTensor<float> convert_from_asymmetric(const SimpleTensor<uint8_t> &src) in convert_from_asymmetric()
39 …SimpleTensor<float> dst{ src.shape(), DataType::F32, 1, QuantizationInfo(), src.data_la… in convert_from_asymmetric()
51 SimpleTensor<float> convert_from_asymmetric(const SimpleTensor<int8_t> &src) in convert_from_asymmetric()
54 …SimpleTensor<float> dst{ src.shape(), DataType::F32, 1, QuantizationInfo(), src.data_la… in convert_from_asymmetric()
67 SimpleTensor<float> convert_from_asymmetric(const SimpleTensor<uint16_t> &src) in convert_from_asymmetric()
70 …SimpleTensor<float> dst{ src.shape(), DataType::F32, 1, QuantizationInfo(), src.data_la… in convert_from_asymmetric()
83 SimpleTensor<uint8_t> convert_to_asymmetric(const SimpleTensor<float> &src, const QuantizationInfo … in convert_to_asymmetric()
85 SimpleTensor<uint8_t> dst{ src.shape(), DataType::QASYMM8, 1, quantization_info }; in convert_to_asymmetric()
99 SimpleTensor<int8_t> convert_to_asymmetric(const SimpleTensor<float> &src, const QuantizationInfo &… in convert_to_asymmetric()
101 … SimpleTensor<int8_t> dst{ src.shape(), DataType::QASYMM8_SIGNED, 1, quantization_info }; in convert_to_asymmetric()
[all …]
|
| /external/ComputeLibrary/tests/ |
| D | SimpleTensor.h | 58 class SimpleTensor : public IAccessor
62 SimpleTensor() = default;
69 SimpleTensor(TensorShape shape, Format format);
79 SimpleTensor(TensorShape shape, DataType data_type,
88 SimpleTensor(const SimpleTensor &tensor);
96 SimpleTensor &operator=(SimpleTensor tensor);
98 SimpleTensor(SimpleTensor &&) = default;
100 ~SimpleTensor() = default;
210 friend void swap(SimpleTensor<U> &tensor1, SimpleTensor<U> &tensor2);
223 SimpleTensor<T1> copy_tensor(const SimpleTensor<T2> &tensor) in copy_tensor()
[all …]
|