/aosp_15_r20/external/ComputeLibrary/src/cpu/utils/ |
H A D | CpuAuxTensorHandler.h | 39 class CpuAuxTensorHandler 42 …CpuAuxTensorHandler(int slot_id, TensorInfo &info, ITensorPack &pack, bool pack_inject = false, bo… 73 CpuAuxTensorHandler(TensorInfo &info, ITensor &tensor) in CpuAuxTensorHandler() function 83 CpuAuxTensorHandler(const CpuAuxTensorHandler &) = delete; 84 CpuAuxTensorHandler &operator=(const CpuAuxTensorHandler) = delete; 86 ~CpuAuxTensorHandler() in ~CpuAuxTensorHandler()
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
H A D | CpuWinogradConv2d.cpp | 330 CpuAuxTensorHandler input_nhwc(offset_int_vec(PermutedInput), _input_nhwc, tensors, true); in run() 331 …CpuAuxTensorHandler winograd_input_transformed(offset_int_vec(TransformedInput), _winograd_transfo… in run() 332 … CpuAuxTensorHandler input_workspace(offset_int_vec(WorkspaceIO), _input_workspace, tensors, true); in run() 341 …CpuAuxTensorHandler winograd_output_transformed(offset_int_vec(TransformedOutput), _winograd_trans… in run() 342 …CpuAuxTensorHandler output_workspace(offset_int_vec(WorkspaceIO), _output_workspace, tensors, true… in run() 343 CpuAuxTensorHandler output_nhwc(offset_int_vec(PermutedOutput), _output_nhwc, tensors, true); in run() 348 …CpuAuxTensorHandler winograd_weights_transformed(offset_int_vec(TransformedWeights), _winograd_tra… in run() 381 CpuAuxTensorHandler permuted_weights(_weights_hwio, *weights_aux); in prepare() 397 … CpuAuxTensorHandler winograd_transformed_weights(_winograd_transformed_weights, *weights_transf); in prepare()
|
H A D | CpuGemmLowpMatrixMultiplyCore.cpp | 517 … CpuAuxTensorHandler vector_sum_col(offset_int_vec(VectorSumCol), _vector_sum_col, tensors, false); in run() 518 … CpuAuxTensorHandler vector_sum_row(offset_int_vec(VectorSumRow), _vector_sum_row, tensors, false); in run() 519 CpuAuxTensorHandler tmp_a(offset_int_vec(TmpA), _tmp_a, tensors, false); in run() 520 CpuAuxTensorHandler tmp_b(offset_int_vec(TmpB), _tmp_b, tensors, true); in run() 521 CpuAuxTensorHandler mm_result_s32(offset_int_vec(MMResultS32), _mm_result_s32, tensors, false); in run() 522 CpuAuxTensorHandler signed_a(offset_int_vec(SignedA), _signed_a, tensors, false); in run() 523 CpuAuxTensorHandler signed_output(offset_int_vec(SignedOutput), _signed_output, tensors, false); in run() 685 CpuAuxTensorHandler tmp_b(_tmp_b, *tmp_b_p); in prepare() 698 CpuAuxTensorHandler vector_sum_col(_vector_sum_col, *vector_sum_col_p); in prepare()
|
H A D | CpuSoftmax.cpp | 165 CpuAuxTensorHandler tmp(offset_int_vec(InternalTensorIdx::TMP), _tmp, tensors, true); in run() 166 CpuAuxTensorHandler max(offset_int_vec(InternalTensorIdx::MAX), _max, tensors, true); in run() 168 …CpuAuxTensorHandler input_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_SRC), _input_permute… in run() 169 …CpuAuxTensorHandler output_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_DST), _output_permu… in run()
|
H A D | CpuGemm.cpp | 306 CpuAuxTensorHandler interleaved_a(offset_int_vec(InterleavedLHS), _tmp_a, tensors, true); in run() 307 CpuAuxTensorHandler transposed_b(offset_int_vec(TransposedRHS), _tmp_b, tensors, true); in run() 308 CpuAuxTensorHandler temp_d(offset_int_vec(TempResult), _tmp_d, tensors, true); in run() 368 CpuAuxTensorHandler transposed_b(_tmp_b, *b_aux); in prepare()
|
H A D | CpuFullyConnected.cpp | 464 CpuAuxTensorHandler flattened_src(offset_int_vec(FlattenedSrc), _flattened_src, tensors, false); in run() 465 …CpuAuxTensorHandler transformed_wei(offset_int_vec(_trans_weights_idx), _trans_weights, tensors, f… in run() 498 …CpuAuxTensorHandler reshaped_weights(offset_int_vec(TransposedWeights), _reshaped_weights, tensors… in prepare() 499 …CpuAuxTensorHandler converted_weights(offset_int_vec(ConvertedWeights), _converted_weights, tensor… in prepare()
|
H A D | CpuAddMulAdd.cpp | 97 …CpuAuxTensorHandler dequantized_bn_mul_handler(offset_int_vec(DequantizedBnMul), _dequantized_bn_m… in run() 98 …CpuAuxTensorHandler dequantized_bn_add_handler(offset_int_vec(DequantizedBnAdd), _dequantized_bn_a… in run()
|
H A D | CpuGemmConv2d.cpp | 557 CpuAuxTensorHandler im2col_output(offset_int_vec(Im2ColOutput), _im2col_output, tensors, false); in run() 558 CpuAuxTensorHandler gemm_output(offset_int_vec(GemmOutput), _gemm_output, tensors, false); in run() 559 …CpuAuxTensorHandler reshaped_wei(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors, fals… in run() 658 … CpuAuxTensorHandler weights_reshaped(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors); in prepare()
|
H A D | CpuGemmDirectConv2d.cpp | 224 CpuAuxTensorHandler permuted_weights(_perm_weights, *weights_aux); in prepare()
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/internal/ |
H A D | CpuGemmAssemblyDispatch.cpp | 437 …CpuAuxTensorHandler pretranspose(offset_int_vec(Pretranspose), _pretranspose_info, tensors, false); in prepare() 516 … CpuAuxTensorHandler pretranspose(offset_int_vec(Pretranspose), _pretranspose_info, tensors, true); in run() 533 … CpuAuxTensorHandler workspace(offset_int_vec(AsmGemmWorkspace), _workspace_info, tensors, false); in run()
|