Home
last modified time | relevance | path

Searched refs:get_softmax_output_quantization_info (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuSoftmaxKernel.cpp225 …put_quantization = is_quantized_asymmetric ? arm_compute::get_softmax_output_quantization_info(src… in validate_arguments_logits_softmax()
261 …put_quantization = is_quantized_asymmetric ? arm_compute::get_softmax_output_quantization_info(src… in configure()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClSoftmaxKernel.cpp128 …const QuantizationInfo allowed_quantization_info = get_softmax_output_quantization_info(info.input… in validate_arguments_1DNorm()
275 …const QuantizationInfo allowed_quantization_info = get_softmax_output_quantization_info(inf… in configure()
/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DSoftmaxLayerNode.cpp67 out_desc.quant_info = get_softmax_output_quantization_info(out_desc.data_type, false); in configure_output()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DSoftmaxLayerFixture.h88 …Type dst = create_tensor<TensorType>(shape, data_type, 1, get_softmax_output_quantization_info(dat… in compute_target()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DSoftmaxLayer.cpp118 … QuantizationInfo output_quantization_info = arm_compute::get_softmax_output_quantization_info(src… in softmax_layer()
/aosp_15_r20/external/ComputeLibrary/src/core/
H A DUtils.cpp537 QuantizationInfo get_softmax_output_quantization_info(DataType input_type, bool is_log) in get_softmax_output_quantization_info() function
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/
H A DUtils.h830 QuantizationInfo get_softmax_output_quantization_info(DataType input_type, bool is_log);