| /external/ComputeLibrary/tests/validation/reference/ |
| D | ReductionOperation.cpp | 41 OT reduce_operation(const T *ptr, int reduce_elements, ReductionOperation op, int stride, RoundingP… in reduce_operation()
47 case ReductionOperation::PROD: in reduce_operation()
52 case ReductionOperation::MIN: in reduce_operation()
53 case ReductionOperation::MAX: in reduce_operation()
73 case ReductionOperation::MIN: in reduce_operation()
79 case ReductionOperation::MAX: in reduce_operation()
85 case ReductionOperation::SUM_SQUARE: in reduce_operation()
88 case ReductionOperation::MEAN_SUM: in reduce_operation()
89 case ReductionOperation::SUM: in reduce_operation()
92 case ReductionOperation::PROD: in reduce_operation()
[all …]
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | ArgMinMax.cpp | 63 …ke("Operation", { ReductionOperation::ARG_IDX_MAX, ReductionOperation::ARG_IDX_MAX, ReductionOpera…
80 … 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOpera…
89 … 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOpera…
102 … 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOpera… in TEST_SUITE()
111 … 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOpera…
123 … 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOpera… in TEST_SUITE()
132 … 2, 3 })), framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOpera…
148 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
159 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
172 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
[all …]
|
| D | ReductionOperation.cpp | 57 ReductionOperation::SUM,
58 ReductionOperation::PROD,
59 ReductionOperation::MIN,
60 ReductionOperation::MAX,
78 TEST_SUITE(ReductionOperation) in TEST_SUITE() argument
105 ReductionOperation::SUM_SQUARE,
111 { 0, 1 })), framework::dataset::make("ReductionOperation", {ReductionOperation::SUM,})), KeepDims),
116 …const bool is_arg_min_max = (op == ReductionOperation::ARG_IDX_MAX) || (op == ReductionOperation::…
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | ArgMinMax.cpp | 78 … { ReductionOperation::ARG_IDX_MAX, ReductionOperation::ARG_IDX_MAX, ReductionOperation::ARG_IDX_M…
96 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
105 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
118 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
128 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
140 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
150 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
167 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
178 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
191 …framework::dataset::make("Operation", { ReductionOperation::ARG_IDX_MIN, ReductionOperation::ARG_I…
[all …]
|
| D | ReductionOperation.cpp | 55 ReductionOperation::SUM,
56 ReductionOperation::PROD,
57 ReductionOperation::MEAN_SUM
62 ReductionOperation::MIN,
63 ReductionOperation::MAX,
70 TEST_SUITE(ReductionOperation) in TEST_SUITE() argument
100 ReductionOperation::SUM_SQUARE,
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEReductionOperationKernel.cpp | 64 uint32x4x4_t calculate_index(uint32_t idx, T a, T b, uint32x4x4_t c, ReductionOperation op, int axi… in calculate_index()
67 if(op == ReductionOperation::ARG_IDX_MIN) in calculate_index()
87 uint32x4x4_t calculate_index_quantized(uint32_t idx, T a, T b, uint32x4x4_t c, ReductionOperation o… in calculate_index_quantized()
91 if(op == ReductionOperation::ARG_IDX_MIN) in calculate_index_quantized()
177 uint32_t calculate_vector_index(uint32x4x4_t vec_res_idx, T vec_res_value, ReductionOperation op) in calculate_vector_index()
182 if(op == ReductionOperation::ARG_IDX_MIN) in calculate_vector_index()
204 … calculate_vector_index_quantized(uint32x4x4_t vec_res_idx, T vec_res_value, ReductionOperation op) in calculate_vector_index_quantized()
209 if(op == ReductionOperation::ARG_IDX_MIN) in calculate_vector_index_quantized()
252 …_index(uint32_t idx, float16x8_t a, float16x8_t b, uint32x4x4_t c, ReductionOperation op, int axis) in calculate_index()
256 if(op == ReductionOperation::ARG_IDX_MIN) in calculate_index()
[all …]
|
| D | NEReductionOperationKernel.h | 68 void configure(const ITensor *input, ITensor *output, unsigned int axis, ReductionOperation op);
80 …ate(const ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op);
89 ReductionOperation _op;
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLReductionOperationKernel.cpp | 45 …ents(const ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op) in validate_arguments()
58 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(op == ReductionOperation::SUM_SQUARE && input->data_type() == Data… in validate_arguments()
61 …ARM_COMPUTE_RETURN_ERROR_ON((op == ReductionOperation::MEAN_SUM) && (axis == 0) && (input->dimensi… in validate_arguments()
63 …ARM_COMPUTE_RETURN_ERROR_ON_MSG((op == ReductionOperation::ARG_IDX_MAX) || (op == ReductionOperati… in validate_arguments()
76 : _input(nullptr), _output(nullptr), _reduction_axis(0), _op(ReductionOperation::SUM_SQUARE) in CLReductionOperationKernel()
81 …nel::configure(const ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op) in configure()
86 …mpile_context, const ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op) in configure()
126 build_opts.add_option_if(op == ReductionOperation::SUM_SQUARE, "-DSUM_SQUARE"); in configure()
127 build_opts.add_option_if(op == ReductionOperation::MEAN_SUM, "-DMEAN"); in configure()
128 build_opts.add_option_if(op == ReductionOperation::SUM, "-DSUM"); in configure()
[all …]
|
| D | CLReductionOperationKernel.h | 60 …void configure(const ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op…
70 …pile_context, const ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op);
82 …ate(const ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op);
91 ReductionOperation _op;
|
| D | CLArgMinMaxLayerKernel.h | 67 …*input, const ICLTensor *prev_output, ICLTensor *output, unsigned int axis, ReductionOperation op);
79 …*input, const ICLTensor *prev_output, ICLTensor *output, unsigned int axis, ReductionOperation op);
93 …nst ITensorInfo *prev_output, const ITensorInfo *output, unsigned int axis, ReductionOperation op);
103 ReductionOperation _op;
|
| D | CLArgMinMaxLayerKernel.cpp | 43 …onst ITensorInfo *prev_output, const ITensorInfo *output, unsigned int axis, ReductionOperation op) in validate_arguments()
48 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(op != ReductionOperation::ARG_IDX_MAX && op != ReductionOperation:… in validate_arguments()
70 …, _prev_output(nullptr), _output(nullptr), _reduction_axis(0), _op(ReductionOperation::ARG_IDX_MAX) in CLArgMinMaxLayerKernel()
75 … *input, const ICLTensor *prev_output, ICLTensor *output, unsigned int axis, ReductionOperation op) in configure()
80 … *input, const ICLTensor *prev_output, ICLTensor *output, unsigned int axis, ReductionOperation op) in configure()
107 build_opts.add_option_if_else(op == ReductionOperation::ARG_IDX_MAX, "-DARG_MAX", "-DARG_MIN"); in configure()
149 …onst ITensorInfo *prev_output, const ITensorInfo *output, unsigned int axis, ReductionOperation op) in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEArgMinMaxLayer.cpp | 44 …EArgMinMaxLayer::configure(ITensor *input, int axis, ITensor *output, const ReductionOperation &op) in configure()
50 …lidate(const ITensorInfo *input, int axis, const ITensorInfo *output, const ReductionOperation &op) in validate()
52 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(op != ReductionOperation::ARG_IDX_MAX && op != ReductionOperation:… in validate()
|
| D | NEReductionOperation.cpp | 66 … ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op, bool kee… in validate()
87 …const auto is_arg_min_max = (op == ReductionOperation::ARG_IDX_MAX) || (op == ReductionOperati… in validate()
105 …ion::configure(ITensor *input, ITensor *output, unsigned int axis, ReductionOperation op, bool kee… in configure()
113 …const auto is_arg_min_max = (op == ReductionOperation::ARG_IDX_MAX) || (op == ReductionOperation:… in configure()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ReductionOperationFixture.h | 49 …void setup(TensorShape shape, DataType data_type, unsigned int axis, ReductionOperation op, Quanti…
51 …const bool is_arg_min_max = (op == ReductionOperation::ARG_IDX_MAX) || (op == ReductionOperation::…
101 …nst TensorShape &src_shape, DataType data_type, unsigned int axis, ReductionOperation op, Quantiza… in compute_target()
130 …nst TensorShape &dst_shape, DataType data_type, unsigned int axis, ReductionOperation op, Quantiza… in compute_reference()
153 …void setup(TensorShape shape, DataType data_type, unsigned int axis, ReductionOperation op, Quanti…
164 …void setup(TensorShape shape, DataType data_type, unsigned int axis, ReductionOperation op, bool k…
|
| D | ArgMinMaxFixture.h | 50 …void setup(TensorShape shape, DataType data_type, int axis, ReductionOperation op, QuantizationInf… in setup()
101 …mpute_target(TensorShape &src_shape, DataType data_type, int axis, ReductionOperation op, Quantiza… in compute_target()
130 …te_reference(TensorShape &src_shape, DataType data_type, int axis, ReductionOperation op, Quantiza… in compute_reference()
151 …void setup(const TensorShape &shape, DataType data_type, int axis, ReductionOperation op, Quantiza… in setup()
162 void setup(const TensorShape &shape, DataType data_type, int axis, ReductionOperation op) in setup()
|
| /external/ComputeLibrary/arm_compute/graph/nodes/ |
| D | ReductionLayerNode.h | 38 ReductionLayerNode(ReductionOperation op, unsigned int axis, bool keep_dims = true);
43 ReductionOperation op() const;
62 ReductionOperation _op;
|
| D | ArgMinMaxLayerNode.h | 44 ArgMinMaxLayerNode(ReductionOperation op,
52 ReductionOperation reduction_operation() const;
74 ReductionOperation _op;
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClArgMinMaxWorkload.cpp | 45 … arm_compute::ReductionOperation::ARG_IDX_MAX); in ClArgMinMaxWorkloadValidate()
50 … arm_compute::ReductionOperation::ARG_IDX_MIN); in ClArgMinMaxWorkloadValidate()
81 arm_compute::ReductionOperation::ARG_IDX_MAX); in ClArgMinMaxWorkload()
89 arm_compute::ReductionOperation::ARG_IDX_MIN); in ClArgMinMaxWorkload()
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonArgMinMaxWorkload.cpp | 45 … arm_compute::ReductionOperation::ARG_IDX_MAX); in NeonArgMinMaxWorkloadValidate()
50 … arm_compute::ReductionOperation::ARG_IDX_MIN); in NeonArgMinMaxWorkloadValidate()
76 layer->configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MAX); in NeonArgMinMaxWorkload()
80 layer->configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MIN); in NeonArgMinMaxWorkload()
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLArgMinMaxLayer.h | 85 … void configure(const ICLTensor *input, int axis, ICLTensor *output, const ReductionOperation &op);
94 …ompile_context, const ICLTensor *input, int axis, ICLTensor *output, const ReductionOperation &op);
104 …idate(const ITensorInfo *input, int axis, const ITensorInfo *output, const ReductionOperation &op);
|
| D | CLReductionOperation.h | 83 …void configure(ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op, bool…
93 …e_context, ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op, bool kee…
105 … ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op, bool kee…
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLArgMinMaxLayer.cpp | 48 …lidate(const ITensorInfo *input, int axis, const ITensorInfo *output, const ReductionOperation &op) in validate()
53 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(op != ReductionOperation::ARG_IDX_MAX && op != ReductionOperation:… in validate()
116 …Layer::configure(const ICLTensor *input, int axis, ICLTensor *output, const ReductionOperation &op) in configure()
121 …compile_context, const ICLTensor *input, int axis, ICLTensor *output, const ReductionOperation &op) in configure()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEArgMinMaxLayer.h | 84 void configure(ITensor *input, int axis, ITensor *output, const ReductionOperation &op);
94 …idate(const ITensorInfo *input, int axis, const ITensorInfo *output, const ReductionOperation &op);
|
| D | NEReductionOperation.h | 79 …void configure(ITensor *input, ITensor *output, unsigned int axis, ReductionOperation op, bool kee…
91 … ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op, bool kee…
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | ReductionLayerNode.cpp | 35 ReductionLayerNode::ReductionLayerNode(ReductionOperation op, unsigned int axis, bool keep_dims) in ReductionLayerNode()
42 ReductionOperation ReductionLayerNode::op() const in op()
|