MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / needs_serialized_reduction

Function needs_serialized_reduction

src/core/Utils.cpp:412–419  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

410}
411
412bool needs_serialized_reduction(ReductionOperation op, DataType dt, unsigned int axis)
413{
414 const bool is_min_max = (op == ReductionOperation::MAX || op == ReductionOperation::MIN);
415 const bool is_quantized_type = is_data_type_quantized(dt);
416 const bool is_first_dim = (axis == 0);
417
418 return !is_first_dim || (is_quantized_type && !is_min_max);
419}
420
421QuantizationInfo get_softmax_output_quantization_info(DataType input_type, bool is_log)
422{

Callers 2

configureMethod · 0.85
runMethod · 0.85

Calls 1

is_data_type_quantizedFunction · 0.85

Tested by

no test coverage detected