Method SplitPlugin

plugin/splitPlugin/split.h:70–75

public:
    SplitPlugin(int32_t axis, int32_t* const& output_lengths, int32_t noutput)
        : _axis(axis)
        , _output_lengths(std::vector<int32_t>(output_lengths, output_lengths + noutput))
    {
        PLUGIN_ASSERT(axis <= nvinfer1::Dims::MAX_DIMS);
    }
    SplitPlugin(int32_t axis, std::vector<int32_t> output_lengths)
        : _axis(axis)
        , _output_lengths(output_lengths)