Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuSoftmax.cpp134 …const TensorInfo tensor_info_max_sum(src->clone()->set_tensor_shape(max_sum_shape).set_data_type(t… in validate() local
151 ARM_COMPUTE_RETURN_ON_ERROR(kernels::CpuLogits1DMaxKernel::validate(src, &tensor_info_max_sum)); in validate()
152 …els::CpuLogits1DSoftmaxKernel<IS_LOG>::validate(&tensor_info_tmp, &tensor_info_max_sum, dst, beta,… in validate()