Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/cpu/operators/
DCpuDepthwiseConv2dAssemblyDispatch.cpp39 struct CpuDepthwiseConv2dAssemblyDispatch::LocalImpl
48 CpuDepthwiseConv2dAssemblyDispatch::CpuDepthwiseConv2dAssemblyDispatch() in CpuDepthwiseConv2dAssemblyDispatch() function in arm_compute::cpu::CpuDepthwiseConv2dAssemblyDispatch
54 CpuDepthwiseConv2dAssemblyDispatch::~CpuDepthwiseConv2dAssemblyDispatch() = default;
56 void CpuDepthwiseConv2dAssemblyDispatch::configure(const ITensorInfo *src, in configure()
69 if(!CpuDepthwiseConv2dAssemblyDispatch::validate(src, weights, bias, dst, info)) in configure()
85 Status CpuDepthwiseConv2dAssemblyDispatch::validate(const ITensorInfo *src, const ITensorInfo *weig… in validate()
90 experimental::MemoryRequirements CpuDepthwiseConv2dAssemblyDispatch::workspace() const in workspace()
95 bool CpuDepthwiseConv2dAssemblyDispatch::is_activation_supported(const ActivationLayerInfo &activat… in is_activation_supported()
101 void CpuDepthwiseConv2dAssemblyDispatch::run(ITensorPack &tensors) in run()
110 void CpuDepthwiseConv2dAssemblyDispatch::prepare(ITensorPack &tensors) in prepare()
DCpuDepthwiseConv2dAssemblyDispatch.h35 class CpuDepthwiseConv2dAssemblyDispatch : public ICpuOperator
38 CpuDepthwiseConv2dAssemblyDispatch();
39 ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuDepthwiseConv2dAssemblyDispatch);
40 ~CpuDepthwiseConv2dAssemblyDispatch();
DCpuDepthwiseConv2d.cpp64 …ARM_COMPUTE_RETURN_ON_ERROR(CpuDepthwiseConv2dAssemblyDispatch::validate(src, weights, biases, dst… in validate_arguments_optimized()
67 …if(info.act_info.enabled() && !CpuDepthwiseConv2dAssemblyDispatch::is_activation_supported(info.ac… in validate_arguments_optimized()
94 …_is_activationlayer_enabled = info.act_info.enabled() && !CpuDepthwiseConv2dAssemblyDispatch::is_a… in configure()
96 _dwc_optimized_func = std::make_unique<CpuDepthwiseConv2dAssemblyDispatch>(); in configure()
363 …if(info.act_info.enabled() && !CpuDepthwiseConv2dAssemblyDispatch::is_activation_supported(info.ac… in validate()
DCpuDepthwiseConv2d.h135 std::unique_ptr<CpuDepthwiseConv2dAssemblyDispatch> _dwc_optimized_func{ nullptr };
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEDepthwiseConvolutionLayer.cpp101 auto dwc_optimized_func = std::make_unique<cpu::CpuDepthwiseConv2dAssemblyDispatch>(); in configure()
/external/ComputeLibrary/src/
DCMakeLists.txt817 cpu/operators/CpuDepthwiseConv2dAssemblyDispatch.cpp
DBUILD.bazel825 "cpu/operators/CpuDepthwiseConv2dAssemblyDispatch.cpp",
/external/ComputeLibrary/
DAndroid.bp472 "src/cpu/operators/CpuDepthwiseConv2dAssemblyDispatch.cpp",