Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuSoftmax.cpp84 TensorInfo tensor_info_tmp(input_info.clone()->set_data_type(tmp_data_type)); in configure() local
89 _tmp = TensorInfo(tensor_info_tmp); in configure()
130 …const TensorInfo tensor_info_tmp(src->clone()->set_data_type(tmp_data_type).set_is_resizable(true)… in validate() local
152 …RETURN_ON_ERROR(kernels::CpuLogits1DSoftmaxKernel<IS_LOG>::validate(&tensor_info_tmp, &tensor_info… in validate()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClSoftmax.cpp120 TensorInfo tensor_info_tmp(src.clone()->set_data_type(tmp_data_type).set_is_resizable(true)); in validate() local
127 …nels::ClLogits1DMaxShiftExpSumKernel::validate(src, tensor_info_max, tensor_info_tmp, tensor_info_… in validate()
128 …ARM_COMPUTE_RETURN_ON_ERROR(kernels::ClLogits1DNormKernel::validate(tensor_info_tmp, tensor_info_s… in validate()