Method parse

src/onnx/parse_split.cpp:166–193

    std::vector<op_desc> operators() const { return {{"Split"}}; }

    std::vector<instruction_ref> parse(const op_desc& opd,
                                       const onnx_parser& parser,
                                       onnx_parser::node_info info,
                                       std::vector<instruction_ref> args) const
    {
        int64_t axis = 0;
        if(contains(info.attributes, "axis"))
        {
            axis = parser.parse_value(info.attributes.at("axis")).at<int>();
        }

        const auto& input_shape = args[0]->get_shape();
        // axis over which the split occurs (split_axis)
        int64_t tuned_axis = tune_axis(input_shape.ndim(), axis, opd.onnx_name);

        auto split_axis_is_fixed = [&]() {
            return input_shape.dyn_dims().at(tuned_axis).is_fixed();
        };

        if(input_shape.dynamic() and not split_axis_is_fixed())
        {
            return parse_dyn_split(info, args, tuned_axis);
        }
        else
        {
            return parse_static_split(info, parser, args, tuned_axis);
        }
    }
};

} // namespace onnx