65 lines
2.6 KiB
C++
65 lines
2.6 KiB
C++
|
// Copyright (C) 2018-2025 Intel Corporation
|
||
|
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
|
//
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "convolution_base.hpp"
|
||
|
|
|
||
|
|
namespace ov {
|
||
|
|
namespace op {
|
||
|
|
namespace util {
|
||
|
|
/// \brief Base class for operations like back propagation convolution
|
||
|
|
class OPENVINO_API ConvolutionBackPropBase : public ConvolutionBase {
|
||
|
|
public:
|
||
|
|
OPENVINO_OP("ConvolutionBackPropBase", "util", ConvolutionBase);
|
||
|
|
|
||
|
|
/// \brief Constructs a conversion operation.
|
||
|
|
ConvolutionBackPropBase() = default;
|
||
|
|
|
||
|
|
/// \brief Constructs a conversion operation.
|
||
|
|
/// \param strides Convolution strides.
|
||
|
|
/// \param pads_begin Amount of padding to be added to the beginning along
|
||
|
|
/// each axis. For example in case of a 2D input the value
|
||
|
|
/// of (1, 2) means that 1 element will be added to the
|
||
|
|
/// top and 2 elements to the left.
|
||
|
|
/// \param pads_end Amount of padding to be added to the end along each
|
||
|
|
/// axis.
|
||
|
|
/// \param dilations The distance in width and height between the weights
|
||
|
|
/// in the filters tensor.
|
||
|
|
/// \param auto_pad Specifies how the automatic calculation of padding
|
||
|
|
/// should be done.
|
||
|
|
/// \param output_padding The output padding adds additional amount of paddings per
|
||
|
|
/// each spatial axis in the output tensor. clang-format on
|
||
|
|
ConvolutionBackPropBase(const OutputVector& arguments,
|
||
|
|
const Strides& strides,
|
||
|
|
const CoordinateDiff& pads_begin,
|
||
|
|
const CoordinateDiff& pads_end,
|
||
|
|
const Strides& dilations,
|
||
|
|
const PadType& auto_pad = PadType::EXPLICIT,
|
||
|
|
const CoordinateDiff& output_padding = {})
|
||
|
|
: ConvolutionBase(arguments, strides, pads_begin, pads_end, dilations, auto_pad),
|
||
|
|
m_output_padding{output_padding} {}
|
||
|
|
|
||
|
|
const CoordinateDiff& get_output_padding() const {
|
||
|
|
return m_output_padding;
|
||
|
|
}
|
||
|
|
void set_output_padding(const CoordinateDiff& output_padding) {
|
||
|
|
m_output_padding = output_padding;
|
||
|
|
}
|
||
|
|
|
||
|
|
protected:
|
||
|
|
CoordinateDiff m_output_padding;
|
||
|
|
|
||
|
|
void resize_attributes(size_t num_spatial) {
|
||
|
|
ConvolutionBase::resize_attributes(num_spatial);
|
||
|
|
|
||
|
|
if (m_output_padding.empty()) {
|
||
|
|
m_output_padding.resize(num_spatial, 0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
} // namespace util
|
||
|
|
} // namespace op
|
||
|
|
} // namespace ov
|