Method PaddingPlugin

stereoDNN/lib/padding_plugin.cpp:18–31

{
public:
    PaddingPlugin(DimsNCHW pad_start, DimsNCHW pad_end, ILogger& log, std::string name):
        pad_start_(pad_start), pad_end_(pad_end), log_(log), name_(name)
    {
        // Only D end padding is currently supported.
        assert(pad_start_.n() == 0);
        assert(pad_start_.c() == 0);
        assert(pad_start_.h() == 0);
        assert(pad_start_.w() == 0);

        assert(pad_end_.n() >= 0);
        assert(pad_end_.c() == 0);
        assert(pad_end_.h() == 0);
        assert(pad_end_.w() == 0);
    }

    PaddingPlugin(PaddingPlugin&&) = delete;