24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_OPERATORS_CPUWINOGRADCONV2D_H
25 #define ACL_SRC_CPU_OPERATORS_CPUWINOGRADCONV2D_H
83 bool enable_fast_math =
false);
96 bool enable_fast_math =
false);
107 TransformedInput = 7,
113 PermutedInput = TransformedOutput,
114 PermutedOutput = TransformedInput
116 std::unique_ptr<CpuGemm> _gemm_function;
117 std::unique_ptr<CpuActivation> _activation_func;
118 std::unique_ptr<ICPPKernel> _transform_input_kernel;
119 std::unique_ptr<ICPPKernel> _transform_output_kernel;
120 std::unique_ptr<CpuPermute> _permute_input;
121 std::unique_ptr<CpuPermute> _permute_output;
122 std::unique_ptr<CpuPermute> _permute_weights;
124 std::unique_ptr<arm_conv::ConvolutionArgs>
126 arm_conv::winograd::WinogradImpl _winograd_impl;
137 bool _run_activation;
142 #endif // ACL_SRC_CPU_OPERATORS_CPUWINOGRADCONV2D_H
void prepare(ITensorPack &constants) override
Prepare the function for executing.
std::vector< MemoryInfo > MemoryRequirements
~CpuWinogradConv2d()
Destructor.
SimpleTensor< float > src
DataLayout
[DataLayout enum definition]
Activation Layer Information class.
void configure(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, ITensorInfo *dst, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo(), bool enable_fast_math=false)
Set the input and output tensors.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Basic interface for functions which have a single async CPU kernel.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
Store the tensor's metadata.
Copyright (c) 2017-2024 Arm Limited.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuWinogradConv2d)
Store the tensor's metadata.
CpuWinogradConv2d()
Constructor.
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo(), bool enable_fast_math=false)
Static function to check if given info will lead to a valid configuration of CpuWinogradConv2d.