| /external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve/ |
| D | integer.cpp | 32 void sve_s32_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window … in sve_s32_elementwise_binary()
36 …_s32_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
37 …_s32_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
38 …_s32_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
39 …_s32_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
40 …_s32_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
41 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
42 …32_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
43 …32_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
46 void sve_s16_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window … in sve_s16_elementwise_binary()
[all …]
|
| D | fp16.cpp | 34 void sve_fp16_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window… in sve_fp16_elementwise_binary()
39 …fp16_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
40 …fp16_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
41 …fp16_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
42 …fp16_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
43 …fp16_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
44 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
45 …16_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
46 …16_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
49 void sve_fp16_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, c… in sve_fp16_comparison_elementwise_binary()
[all …]
|
| D | fp32.cpp | 32 void sve_fp32_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window… in sve_fp32_elementwise_binary()
37 …fp32_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
38 …fp32_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
39 …fp32_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
40 …fp32_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
41 …fp32_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
42 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
43 …32_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
44 …32_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
47 void sve_fp32_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, c… in sve_fp32_comparison_elementwise_binary()
[all …]
|
| /external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/neon/ |
| D | integer.cpp | 31 void neon_s32_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window… in neon_s32_elementwise_binary()
36 …_s32_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
37 …_s32_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
38 …_s32_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
39 …_s32_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
40 …_s32_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
41 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
42 …32_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
43 …32_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
46 void neon_s16_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Window… in neon_s16_elementwise_binary()
[all …]
|
| D | qasymm8.cpp | 31 void neon_qasymm8_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Wi… in neon_qasymm8_elementwise_binary()
36 …ymm8_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
37 …ymm8_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
38 …ymm8_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
39 …ymm8_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
40 …ymm8_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
41 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
42 …m8_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
43 …m8_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
46 void neon_qasymm8_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *ou… in neon_qasymm8_comparison_elementwise_binary()
[all …]
|
| D | fp32.cpp | 32 void neon_fp32_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Windo… in neon_fp32_elementwise_binary()
37 …fp32_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
38 …fp32_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
39 …fp32_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
40 …fp32_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
41 …fp32_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
42 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
43 …32_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
44 …32_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
47 void neon_fp32_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, … in neon_fp32_comparison_elementwise_binary()
[all …]
|
| D | fp16.cpp | 33 void neon_fp16_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Windo… in neon_fp16_elementwise_binary()
38 …fp16_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
39 …fp16_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
40 …fp16_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
41 …fp16_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
42 …fp16_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
43 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
44 …16_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
45 …16_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
48 void neon_fp16_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, … in neon_fp16_comparison_elementwise_binary()
[all …]
|
| D | qasymm8_signed.cpp | 32 void neon_qasymm8_signed_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, c… in neon_qasymm8_signed_elementwise_binary()
37 …gned_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
38 …gned_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
39 …gned_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
40 …gned_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
41 …gned_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
42 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
43 …ed_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
44 …ed_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
47 …neon_qasymm8_signed_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor … in neon_qasymm8_signed_comparison_elementwise_binary()
[all …]
|
| /external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve2/ |
| D | qasymm8.cpp | 32 void sve2_qasymm8_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, const Wi… in sve2_qasymm8_elementwise_binary()
37 …ymm8_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
38 …ymm8_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
39 …ymm8_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
40 …ymm8_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
41 …ymm8_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
42 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
43 …m8_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
44 …m8_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
47 void sve2_qasymm8_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *ou… in sve2_qasymm8_comparison_elementwise_binary()
[all …]
|
| D | qasymm8_signed.cpp | 32 void sve2_qasymm8_signed_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor *out, c… in sve2_qasymm8_signed_elementwise_binary()
37 …gned_elementwise_binary<ArithmeticOperation::ADD>(const ITensor *in1, const ITensor *in2, ITensor …
38 …gned_elementwise_binary<ArithmeticOperation::SUB>(const ITensor *in1, const ITensor *in2, ITensor …
39 …gned_elementwise_binary<ArithmeticOperation::DIV>(const ITensor *in1, const ITensor *in2, ITensor …
40 …gned_elementwise_binary<ArithmeticOperation::MIN>(const ITensor *in1, const ITensor *in2, ITensor …
41 …gned_elementwise_binary<ArithmeticOperation::MAX>(const ITensor *in1, const ITensor *in2, ITensor …
42 …entwise_binary<ArithmeticOperation::SQUARED_DIFF>(const ITensor *in1, const ITensor *in2, ITensor …
43 …ed_elementwise_binary<ArithmeticOperation::POWER>(const ITensor *in1, const ITensor *in2, ITensor …
44 …ed_elementwise_binary<ArithmeticOperation::PRELU>(const ITensor *in1, const ITensor *in2, ITensor …
47 …sve2_qasymm8_signed_comparison_elementwise_binary(const ITensor *in1, const ITensor *in2, ITensor … in sve2_qasymm8_signed_comparison_elementwise_binary()
[all …]
|
| /external/ComputeLibrary/src/cpu/kernels/select/generic/neon/ |
| D | impl.cpp | 38 void select_op(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const Win… in select_op()
76 void select_op_8(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const W… in select_op_8()
90 void select_op_16(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const … in select_op_16()
104 void select_op_32(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const … in select_op_32()
118 void select_op_not_same_rank(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *… in select_op_not_same_rank()
153 …e void select_op_32<float, uint32x4_t>(const ITensor *cond, const ITensor *in1, const ITensor *in2…
155 …te void select_op_not_same_rank<float>(const ITensor *cond, const ITensor *in1, const ITensor *in2…
157 …e void select_op_8<int8_t, uint8x16_t>(const ITensor *cond, const ITensor *in1, const ITensor *in2…
159 …void select_op_16<int16_t, uint16x8_t>(const ITensor *cond, const ITensor *in1, const ITensor *in2…
162 …id select_op_16<float16_t, uint16x8_t>(const ITensor *cond, const ITensor *in1, const ITensor *in2…
[all …]
|
| D | integer.cpp | 36 void neon_s8_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output… in neon_s8_select_same_rank()
40 void neon_s16_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *outpu… in neon_s16_select_same_rank()
44 void neon_s32_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *outpu… in neon_s32_select_same_rank()
48 void neon_s8_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *ou… in neon_s8_select_not_same_rank()
52 void neon_s16_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *o… in neon_s16_select_not_same_rank()
56 void neon_s32_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *o… in neon_s32_select_not_same_rank()
60 void neon_u8_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *output… in neon_u8_select_same_rank()
64 void neon_u16_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *outpu… in neon_u16_select_same_rank()
68 void neon_u32_select_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *outpu… in neon_u32_select_same_rank()
72 void neon_u8_select_not_same_rank(const ITensor *c, const ITensor *x, const ITensor *y, ITensor *ou… in neon_u8_select_not_same_rank()
[all …]
|
| D | impl.h | 31 class ITensor; variable
37 void select_op_not_same_rank(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *…
40 void select_op_32(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const …
43 void select_op_16(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const …
46 void select_op_8(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const W…
49 void select_op(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const Win…
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuScaleKernel.h | 42 …ctionPtr = void (CpuScaleKernel::*)(const ITensor *, ITensor *, const ITensor *, const ITensor *, …
43 …KernelPtr = std::add_pointer<void(const ITensor *, ITensor *, const ITensor *, const ITensor *, …
92 …void scale_area_nchw_u8(const ITensor *src, ITensor *dst, const ITensor *dx, const ITensor *dy, co…
96 …void scale_bilinear_nchw(const ITensor *src, ITensor *dst, const ITensor *dx, const ITensor *dy, c…
99 …void scale_bilinear_qasymm(const ITensor *src, ITensor *dst, const ITensor *dx, const ITensor *dy,…
103 …void scale_nearest_nchw(const ITensor *src, ITensor *dst, const ITensor *dx, const ITensor *dy, co…
|
| /external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/ |
| D | list.h | 32 …void func_name(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITen…
33 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
36 …void func_name(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITenso…
37 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
40 …void func_name(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused_weights, ITenso…
41 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEFuseBatchNormalizationKernel.h | 32 class ITensor; variable
69 …void configure(const ITensor *input_weights, const ITensor *bn_mean, const ITensor *bn_var, ITenso…
70 …const ITensor *input_bias = nullptr, const ITensor *bn_beta = nullptr, const ITensor *bn_gamma = n…
98 const ITensor *_input_weights;
99 const ITensor *_input_bias;
100 const ITensor *_bn_mean;
101 const ITensor *_bn_var;
102 const ITensor *_bn_gamma;
103 const ITensor *_bn_beta;
104 ITensor *_fused_weights;
[all …]
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayerQuantized.h | 46 class ITensor; variable
106 void configure(const ITensor *input,
107 …const ITensor *input_to_input_weights, const ITensor *input_to_forget_weights, const ITensor *inpu…
108 …const ITensor *recurrent_to_input_weights, const ITensor *recurrent_to_forget_weights, const ITens…
109 …const ITensor *input_gate_bias, const ITensor *forget_gate_bias, const ITensor *cell_bias, const I…
110 ITensor *cell_state_in, const ITensor *output_state_in,
111 ITensor *cell_state_out, ITensor *output_state_out);
176 const ITensor *_input_to_input_weights;
177 const ITensor *_input_to_forget_weights;
178 const ITensor *_input_to_cell_weights;
[all …]
|
| D | NEQLSTMLayer.h | 45 class ITensor; variable
133 void configure(const ITensor *input,
134 …const ITensor *input_to_forget_weights, const ITensor *input_to_cell_weights, const ITensor *input…
135 …const ITensor *recurrent_to_forget_weights, const ITensor *recurrent_to_cell_weights, const ITenso…
136 … const ITensor *forget_gate_bias, const ITensor *cell_bias, const ITensor *output_gate_bias,
137 const ITensor *cell_state_in, ITensor *output_state_in,
138 ITensor *cell_state_out, ITensor *output_state_out, ITensor *output,
139 const LSTMParams<ITensor> &lstm_params);
222 … const ITensor *mm_input, const ITensor *mm_weights, const ITensor *bias, Tensor *mm_res,
233 ITensor *_src{ nullptr };
[all …]
|
| /external/ComputeLibrary/arm_compute/core/CPP/kernels/ |
| D | CPPBoxWithNonMaximaSuppressionLimitKernel.h | 32 class ITensor; variable
66 …nfigure(const ITensor *scores_in, const ITensor *boxes_in, const ITensor *batch_splits_in, ITensor…
67 …ITensor *batch_splits_out = nullptr, ITensor *keeps = nullptr, ITensor *keeps_size = nullptr, cons…
77 const ITensor *_scores_in;
78 const ITensor *_boxes_in;
79 const ITensor *_batch_splits_in;
80 ITensor *_scores_out;
81 ITensor *_boxes_out;
82 ITensor *_classes;
83 ITensor *_batch_splits_out;
[all …]
|
| /external/ComputeLibrary/arm_compute/runtime/CPP/functions/ |
| D | CPPBoxWithNonMaximaSuppressionLimit.h | 36 class ITensor; variable
64 …nfigure(const ITensor *scores_in, const ITensor *boxes_in, const ITensor *batch_splits_in, ITensor…
65 …ITensor *batch_splits_out = nullptr, ITensor *keeps = nullptr, ITensor *keeps_size = nullptr, cons…
95 const ITensor *_scores_in;
96 const ITensor *_boxes_in;
97 const ITensor *_batch_splits_in;
98 ITensor *_scores_out;
99 ITensor *_boxes_out;
100 ITensor *_classes;
101 ITensor *_batch_splits_out;
[all …]
|
| D | CPPDetectionPostProcessLayer.h | 39 class ITensor; variable
68 …void configure(const ITensor *input_box_encoding, const ITensor *input_score, const ITensor *input…
69 …ITensor *output_boxes, ITensor *output_classes, ITensor *output_scores, ITensor *num_detection, De…
92 const ITensor *_input_box_encoding;
93 const ITensor *_input_scores;
94 const ITensor *_input_anchors;
95 ITensor *_output_boxes;
96 ITensor *_output_classes;
97 ITensor *_output_scores;
98 ITensor *_num_detection;
[all …]
|
| /external/ComputeLibrary/src/cpu/kernels/maxunpool/generic/neon/ |
| D | impl.cpp | 27 class ITensor;
32 void max_unpooling(const ITensor *input, const ITensor *indices, ITensor *output, const Window &win… in max_unpooling()
46 template void max_unpooling<float>(const ITensor *input, const ITensor *indices, ITensor *output, c…
47 template void max_unpooling<int8_t>(const ITensor *input, const ITensor *indices, ITensor *output, …
48 template void max_unpooling<uint8_t>(const ITensor *input, const ITensor *indices, ITensor *output,…
51 template void max_unpooling<float16_t>(const ITensor *input, const ITensor *indices, ITensor *outpu…
|
| /external/ComputeLibrary/src/cpu/kernels/add/generic/neon/ |
| D | impl.h | 34 void add_same_neon(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPolicy &pol…
42 void add_sub_qasymm8_neon(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPoli…
44 void add_sub_qasymm8_signed_neon(const ITensor *src0, const ITensor *src1, ITensor *dst, const Conv…
47 void add_q8_neon_fixedpoint(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPo…
50 void add_sub_q8_neon_fixedpoint(const ITensor *src0, const ITensor *src1, ITensor *dst, const Conve…
|
| /external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/generic/ |
| D | impl.cpp | 31 …used_batch_normalization_conv(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fuse… in fused_batch_normalization_conv()
32 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_conv()
126 …normalization_conv<float32_t>(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fuse…
127 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
130 …normalization_conv<float16_t>(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fuse…
131 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl…
|
| /external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nchw/ |
| D | all.cpp | 32 …d_batch_normalization_dwc_nchw(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused… in fused_batch_normalization_dwc_nchw()
33 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nchw()
128 …tch_normalization_dwc_nchw_f32(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused… in fused_batch_normalization_dwc_nchw_f32()
129 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nchw_f32()
136 …tch_normalization_dwc_nchw_f16(const ITensor *dwc_weights, const ITensor *dwc_bias, ITensor *fused… in fused_batch_normalization_dwc_nchw_f16()
137 …const ITensor *bn_mean, const ITensor *bn_var, const ITensor *bn_beta, const ITensor *bn_gamma, fl… in fused_batch_normalization_dwc_nchw_f16()
|