Home
last modified time | relevance | path

Searched refs:get_const_tensor (Results 1 – 25 of 152) sorted by relevance

1234567

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuWinogradConv2dKernel.cpp39 const ITensor *input_nhwc = tensors.get_const_tensor(TensorType::ACL_SRC); in run_op()
40 const ITensor *winograd_input_transform = tensors.get_const_tensor(TensorType::ACL_DST); in run_op()
41 const ITensor *workspace = tensors.get_const_tensor(TensorType::ACL_INT); in run_op()
77 const ITensor *dst_nhwc = tensors.get_const_tensor(TensorType::ACL_DST); in run_op()
78 const ITensor *winograd_output_transform = tensors.get_const_tensor(TensorType::ACL_SRC_0); in run_op()
79 const ITensor *biases = tensors.get_const_tensor(TensorType::ACL_SRC_1); in run_op()
H A DCpuAddMulAddKernel.cpp183 const ITensor *input1 = tensors.get_const_tensor(TensorType::ACL_SRC_0); in run_op()
184 const ITensor *input2 = tensors.get_const_tensor(TensorType::ACL_SRC_1); in run_op()
185 const ITensor *bn_mul = tensors.get_const_tensor(TensorType::ACL_SRC_2); in run_op()
186 const ITensor *bn_add = tensors.get_const_tensor(TensorType::ACL_SRC_3); in run_op()
H A DCpuDirectConv3dKernel.cpp171 auto src0 = tensors.get_const_tensor(TensorType::ACL_SRC_0); in run_op()
172 auto src1 = tensors.get_const_tensor(TensorType::ACL_SRC_1); in run_op()
173 auto src2 = tensors.get_const_tensor(TensorType::ACL_SRC_2); in run_op()
H A DCpuDepthwiseConv2dNativeKernel.cpp179 const auto src = tensors.get_const_tensor(TensorType::ACL_SRC_0); in run_op()
180 const auto weights = tensors.get_const_tensor(TensorType::ACL_SRC_1); in run_op()
181 const auto biases = tensors.get_const_tensor(TensorType::ACL_SRC_2); in run_op()
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuDepthwiseConv2d.cpp152 auto bias = tensors.get_const_tensor(TensorType::ACL_SRC_2); in run()
161 auto src = tensors.get_const_tensor(TensorType::ACL_SRC_0); in run()
226 auto weights = tensors.get_const_tensor(TensorType::ACL_SRC_1); in prepare()
227 auto bias = tensors.get_const_tensor(TensorType::ACL_SRC_2); in prepare()
243 auto weights = tensors.get_const_tensor(TensorType::ACL_SRC_1); in prepare()
244 auto bias = tensors.get_const_tensor(TensorType::ACL_SRC_2); in prepare()
373 auto src = tensors.get_const_tensor(TensorType::ACL_SRC_0); in run()
374 auto weights = tensors.get_const_tensor(TensorType::ACL_SRC_1); in run()
375 auto biases = tensors.get_const_tensor(TensorType::ACL_SRC_2); in run()
429 auto weights = tensors.get_const_tensor(TensorType::ACL_SRC_1); in prepare()
H A DCpuAddMulAdd.cpp90 …const DataType data_type = tensors.get_const_tensor(TensorType::ACL_SRC_0)->info()->data_typ… in run()
94 const ITensor *bn_mul = tensors.get_const_tensor(TensorType::ACL_SRC_2); in run()
95 const ITensor *bn_add = tensors.get_const_tensor(TensorType::ACL_SRC_3); in run()
117 { TensorType::ACL_SRC_0, tensors.get_const_tensor(TensorType::ACL_SRC_0) }, in run()
118 { TensorType::ACL_SRC_1, tensors.get_const_tensor(TensorType::ACL_SRC_1) }, in run()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClGemmLowpOffsetContributionOutputStageKernel.cpp224 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
225 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
226 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
227 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
228 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
229 …iers = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
H A DClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp461 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
462 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
463 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
464 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
465 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
466 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
467 …iers = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
H A DClWidthConcatenate4TensorsKernel.cpp167 …const auto src0 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
168 …const auto src1 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
169 …const auto src2 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
170 …const auto src3 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
H A DClIndirectConv2dKernel.cpp269 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
270 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
271 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
272 …ffer = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
H A DClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp423 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
424 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
425 … = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
426 …_col = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
427 …_row = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::A… in run_op()
H A DClGemmMatrixMultiplyReshapedKernel.cpp334 …const auto src0 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
335 …const auto src1 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
336 …const auto src2 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
400 …_arg = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(experimental:… in run_op()
422 …_arg = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(experimental:… in run_op()
H A DClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp339 …const auto src0 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
340 …const auto src1 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
341 …const auto src2 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
415 …_arg = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(experimental:… in run_op()
436 …_arg = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(experimental:… in run_op()
H A DClGemmMatrixMultiplyNativeKernel.cpp372 …const auto src0 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
373 …const auto src1 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
374 …const auto src2 = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(Te… in run_op()
445 …_arg = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(experimental:… in run_op()
458 …_arg = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(experimental:… in run_op()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/
H A DClTemplateDepthwiseConv2d.cpp47 _src = this->tensors().get_const_tensor(TensorType::ACL_SRC_0);
48 _weight = this->tensors().get_const_tensor(TensorType::ACL_SRC_1);
49 if(this->tensors().get_const_tensor(TensorType::ACL_SRC_2))
51 _bias = this->tensors().get_const_tensor(TensorType::ACL_SRC_2);
53 _dst = this->tensors().get_const_tensor(TensorType::ACL_DST_0);
H A DClTemplateDirectConv2d.cpp52 _src = this->tensors().get_const_tensor(TensorType::ACL_SRC_0);
53 _weight = this->tensors().get_const_tensor(TensorType::ACL_SRC_1);
54 if(this->tensors().get_const_tensor(TensorType::ACL_SRC_2))
56 _bias = this->tensors().get_const_tensor(TensorType::ACL_SRC_2);
58 _dst = this->tensors().get_const_tensor(TensorType::ACL_DST_0);
H A DClTemplateLogits1DNorm.cpp46 _src = this->tensors().get_const_tensor(TensorType::ACL_SRC_0);
47 _sum = this->tensors().get_const_tensor(TensorType::ACL_SRC_1);
48 _dst = this->tensors().get_const_tensor(TensorType::ACL_DST_0);
H A DClTemplateLogits1DMaxShiftExpSum.cpp50 _src = this->tensors().get_const_tensor(TensorType::ACL_SRC_0);
51 _sum = this->tensors().get_const_tensor(TensorType::ACL_DST_0);
52 _dst = this->tensors().get_const_tensor(TensorType::ACL_DST_1);
H A DClTemplateElementwiseBinary.cpp51 _lhs = this->tensors().get_const_tensor(TensorType::ACL_SRC_0);
52 _rhs = this->tensors().get_const_tensor(TensorType::ACL_SRC_1);
53 _dst = this->tensors().get_const_tensor(TensorType::ACL_DST_0);
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/
H A DClComponentDirectConv2d.cpp78 const auto src = tensors.get_const_tensor(TensorType::ACL_SRC_0); in validate()
79 const auto wei = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate()
80 const auto bia = tensors.get_const_tensor(TensorType::ACL_SRC_2); in validate()
81 const auto dst = tensors.get_const_tensor(TensorType::ACL_DST_0); in validate()
H A DClComponentDepthwiseConv2d.cpp113 const auto src = tensors.get_const_tensor(TensorType::ACL_SRC_0); in validate()
114 const auto wei = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate()
115 const auto bia = tensors.get_const_tensor(TensorType::ACL_SRC_2); in validate()
116 const auto dst = tensors.get_const_tensor(TensorType::ACL_DST_0); in validate()
H A DClComponentLogits1DMaxShiftExpSum.cpp47 const ITensorInfo *src = tensors.get_const_tensor(TensorType::ACL_SRC_0); in validate()
48 const ITensorInfo *sum = tensors.get_const_tensor(TensorType::ACL_DST_0); in validate()
49 const ITensorInfo *dst = tensors.get_const_tensor(TensorType::ACL_DST_1); in validate()
H A DClComponentLogits1DNorm.cpp47 const ITensorInfo *src = tensors.get_const_tensor(TensorType::ACL_SRC_0); in validate()
48 const ITensorInfo *sum = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate()
49 const ITensorInfo *dst = tensors.get_const_tensor(TensorType::ACL_DST_0); in validate()
H A DClComponentElementwiseBinary.cpp48 const auto lhs = tensors.get_const_tensor(TensorType::ACL_SRC_0); in validate()
49 const auto rhs = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate()
50 const auto dst = tensors.get_const_tensor(TensorType::ACL_DST_0); in validate()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/
H A DArgumentPack.h134 const T *get_const_tensor(Id id) const in get_const_tensor() function
193 auto tensor = get_const_tensor(static_cast<TensorType>(id)); in get_const_src_tensors()
227 auto tensor = get_const_tensor(static_cast<TensorType>(id)); in get_const_dst_tensors()

1234567