Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/cpu/utils/
H A DCpuAuxTensorHandler.h39 class CpuAuxTensorHandler
42CpuAuxTensorHandler(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 DCpuWinogradConv2d.cpp330 CpuAuxTensorHandler input_nhwc(offset_int_vec(PermutedInput), _input_nhwc, tensors, true); in run()
331CpuAuxTensorHandler winograd_input_transformed(offset_int_vec(TransformedInput), _winograd_transfo… in run()
332CpuAuxTensorHandler input_workspace(offset_int_vec(WorkspaceIO), _input_workspace, tensors, true); in run()
341CpuAuxTensorHandler winograd_output_transformed(offset_int_vec(TransformedOutput), _winograd_trans… in run()
342CpuAuxTensorHandler 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()
348CpuAuxTensorHandler winograd_weights_transformed(offset_int_vec(TransformedWeights), _winograd_tra… in run()
381 CpuAuxTensorHandler permuted_weights(_weights_hwio, *weights_aux); in prepare()
397CpuAuxTensorHandler winograd_transformed_weights(_winograd_transformed_weights, *weights_transf); in prepare()
H A DCpuGemmLowpMatrixMultiplyCore.cpp517CpuAuxTensorHandler vector_sum_col(offset_int_vec(VectorSumCol), _vector_sum_col, tensors, false); in run()
518CpuAuxTensorHandler 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 DCpuSoftmax.cpp165 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()
168CpuAuxTensorHandler input_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_SRC), _input_permute… in run()
169CpuAuxTensorHandler output_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_DST), _output_permu… in run()
H A DCpuGemm.cpp306 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 DCpuFullyConnected.cpp464 CpuAuxTensorHandler flattened_src(offset_int_vec(FlattenedSrc), _flattened_src, tensors, false); in run()
465CpuAuxTensorHandler transformed_wei(offset_int_vec(_trans_weights_idx), _trans_weights, tensors, f… in run()
498CpuAuxTensorHandler reshaped_weights(offset_int_vec(TransposedWeights), _reshaped_weights, tensors… in prepare()
499CpuAuxTensorHandler converted_weights(offset_int_vec(ConvertedWeights), _converted_weights, tensor… in prepare()
H A DCpuAddMulAdd.cpp97CpuAuxTensorHandler dequantized_bn_mul_handler(offset_int_vec(DequantizedBnMul), _dequantized_bn_m… in run()
98CpuAuxTensorHandler dequantized_bn_add_handler(offset_int_vec(DequantizedBnAdd), _dequantized_bn_a… in run()
H A DCpuGemmConv2d.cpp557 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()
559CpuAuxTensorHandler reshaped_wei(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors, fals… in run()
658CpuAuxTensorHandler weights_reshaped(offset_int_vec(WeightsReshaped), _weights_reshaped, tensors); in prepare()
H A DCpuGemmDirectConv2d.cpp224 CpuAuxTensorHandler permuted_weights(_perm_weights, *weights_aux); in prepare()
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/internal/
H A DCpuGemmAssemblyDispatch.cpp437CpuAuxTensorHandler pretranspose(offset_int_vec(Pretranspose), _pretranspose_info, tensors, false); in prepare()
516CpuAuxTensorHandler pretranspose(offset_int_vec(Pretranspose), _pretranspose_info, tensors, true); in run()
533CpuAuxTensorHandler workspace(offset_int_vec(AsmGemmWorkspace), _workspace_info, tensors, false); in run()