Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/cpu/operators/internal/
DCpuGemmAssemblyDispatch.cpp166 if(!_gemm_kernel_asm) in isVarWeightsKernel()
168 …:WeightFormat wf = assembly_utils::map_to_arm_compute_weight_format(_gemm_kernel_asm->get_config()… in isVarWeightsKernel()
192 std::shared_ptr<arm_gemm::GemmCommon<TypeInput, TypeOutput>> _gemm_kernel_asm{ nullptr }; member in arm_compute::cpu::__anon91ef70ce0111::Fallback
319 _gemm_kernel_asm->set_convolution_parameters(_cp); in configure_indirect()
352 _gemm_kernel_asm->set_indirect_parameters(a->tensor_shape()[0], _indirect_arg.get()); in configure_indirect()
366 _gemm_kernel_asm = arm_gemm::gemm<TypeInput, TypeOutput, OutputStage>(args, os); in configure()
367 if(_gemm_kernel_asm == nullptr) in configure()
373 arm_gemm::GemmConfig gemm_cfg = _gemm_kernel_asm->get_config(); in configure()
378 acl_gemm_wrapper->configure(_gemm_kernel_asm.get(), gemm_cfg.filter); in configure()
379 const size_t workspace_size = _gemm_kernel_asm->get_working_size(); in configure()
[all …]