Pad Layer node */
| 36 | { |
| 37 | /** Pad Layer node */ |
| 38 | class PadLayerNode final : public INode |
| 39 | { |
| 40 | public: |
| 41 | /** Constructor |
| 42 | * |
| 43 | * @param[in] padding The padding for each spatial dimension of the input tensor. The pair padding[i] |
| 44 | * specifies the front and the end padding in the i-th dimension. |
| 45 | * @param[in] pad_value Padding value to be used. Defaults to 0 |
| 46 | */ |
| 47 | PadLayerNode(const PaddingList &padding, PixelValue pad_value = PixelValue()); |
| 48 | /** Padding list accessor |
| 49 | * |
| 50 | * @return Padding list |
| 51 | */ |
| 52 | const PaddingList &padding() const; |
| 53 | /** Padding value accessor |
| 54 | * |
| 55 | * @return Padding value |
| 56 | */ |
| 57 | PixelValue pad_value() const; |
| 58 | |
| 59 | // Inherited overridden methods: |
| 60 | NodeType type() const override; |
| 61 | bool forward_descriptors() override; |
| 62 | TensorDescriptor configure_output(size_t idx) const override; |
| 63 | void accept(INodeVisitor &v) override; |
| 64 | |
| 65 | public: |
| 66 | static constexpr NodeType node_type = NodeType::PadLayer; |
| 67 | |
| 68 | private: |
| 69 | PaddingList _padding; |
| 70 | PixelValue _pad_value; |
| 71 | }; |
| 72 | } // namespace graph |
| 73 | } // namespace arm_compute |
| 74 | #endif // ACL_ARM_COMPUTE_GRAPH_NODES_PADLAYERNODE_H |