24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_OPERATORS_CPUGEMMDIRECTCONV2D_H
25 #define ACL_SRC_CPU_OPERATORS_CPUGEMMDIRECTCONV2D_H
106 std::unique_ptr<CpuGemmAssemblyDispatch> _gemm_asm_func;
107 std::unique_ptr<CpuActivation> _activation_func;
108 std::unique_ptr<CpuPermute> _weights_permute_func;
111 bool _run_activation;
117 #endif // ACL_SRC_CPU_OPERATORS_CPUGEMMDIRECTCONV2D_H
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
Descriptor used by the 2d Convolution function.
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv2dInfo &info)
Static function to check if given info will lead to a valid configuration of CpuGemmDirectConv2d.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuGemmDirectConv2d)
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Basic interface for functions which have a single async CPU kernel.
Store the tensor's metadata.
void prepare(ITensorPack &constants) override
Prepare the function for executing.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
void configure(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &info)
Set the input and output tensors.