Home
last modified time | relevance | path

Searched refs:CpuGemmAssemblyDispatch (Results 1 – 11 of 11) sorted by relevance

/external/ComputeLibrary/src/cpu/operators/internal/
DCpuGemmAssemblyDispatch.h61 class CpuGemmAssemblyDispatch : public ICpuOperator
65 CpuGemmAssemblyDispatch();
67 ~CpuGemmAssemblyDispatch() = default;
69 ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuGemmAssemblyDispatch);
DCpuGemmAssemblyDispatch.cpp122 class Fallback : public CpuGemmAssemblyDispatch::IFallback
581 void create_arm_gemm(std::unique_ptr<CpuGemmAssemblyDispatch::IFallback> &arm_gemm, in create_arm_gemm()
600 void create_arm_gemm_quant(std::unique_ptr<CpuGemmAssemblyDispatch::IFallback> &arm_gemm, in create_arm_gemm_quant()
647 CpuGemmAssemblyDispatch::CpuGemmAssemblyDispatch() in CpuGemmAssemblyDispatch() function in arm_compute::cpu::CpuGemmAssemblyDispatch
652 Status CpuGemmAssemblyDispatch::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, con… in has_opt_impl()
722 Status CpuGemmAssemblyDispatch::validate(const ITensorInfo *a, const ITensorInfo *b, const ITensorI… in validate()
757 …const Status ret = CpuGemmAssemblyDispatch::has_opt_impl(expected_weight_format, a, b… in validate()
769 bool CpuGemmAssemblyDispatch::is_activation_supported(const ActivationLayerInfo &activation) in is_activation_supported()
775 void CpuGemmAssemblyDispatch::configure(const ITensorInfo *a, const ITensorInfo *b, const ITensorIn… in configure()
781 if(!CpuGemmAssemblyDispatch::validate(a, b, c, d, info)) in configure()
[all …]
/external/ComputeLibrary/src/cpu/operators/
DCpuGemm.cpp68 …bool run_optimised = bool(cpu::CpuGemmAssemblyDispatch::validate(a, b, (is_c_bia… in configure()
77 …n_info().enabled() && (!run_optimised || (run_optimised && !cpu::CpuGemmAssemblyDispatch::is_activ… in configure()
82 _asm_glue = std::make_unique<cpu::CpuGemmAssemblyDispatch>(); in configure()
213 …const bool run_optimised = bool(cpu::CpuGemmAssemblyDispatch::validate(a, b, is_c_bias ? c :… in validate()
386 return CpuGemmAssemblyDispatch::has_opt_impl(expected_weight_format, a, b, c, d, asm_info); in has_opt_impl()
DCpuGemmLowpMatrixMultiplyCore.h48 class CpuGemmAssemblyDispatch; variable
139 std::unique_ptr<CpuGemmAssemblyDispatch> _asm_glue;
DCpuGemmLowpMatrixMultiplyCore.cpp75 : _asm_glue(std::make_unique<CpuGemmAssemblyDispatch>()), in CpuGemmLowpMatrixMultiplyCore()
129 _asm_glue = std::make_unique<cpu::CpuGemmAssemblyDispatch>(); in configure()
281 … = activation.enabled() && (!_assembly_path || !cpu::CpuGemmAssemblyDispatch::is_activ… in configure()
376 …run_optimised = bool(CpuGemmAssemblyDispatch::validate(a_to_use, b, c, output, asm_inf… in validate()
381 …run_optimised = bool(CpuGemmAssemblyDispatch::validate(a_to_use, b, nullptr, fuse_output_stage ? &… in validate()
DCpuGemmDirectConv2d.h95 std::unique_ptr<CpuGemmAssemblyDispatch> _gemm_asm_func;
DCpuGemmDirectConv2d.cpp98 : _gemm_asm_func(std::make_unique<CpuGemmAssemblyDispatch>()), in CpuGemmDirectConv2d()
193 …ARM_COMPUTE_RETURN_ON_ERROR(cpu::CpuGemmAssemblyDispatch::validate(src, weights, biases, dst, asm_… in validate()
DCpuGemm.h143 std::unique_ptr<CpuGemmAssemblyDispatch> _asm_glue{ nullptr };
/external/ComputeLibrary/src/
DCMakeLists.txt844 cpu/operators/internal/CpuGemmAssemblyDispatch.cpp
DBUILD.bazel852 "cpu/operators/internal/CpuGemmAssemblyDispatch.cpp",
/external/ComputeLibrary/
DAndroid.bp499 "src/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp",