| /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()
|