class ov::pass::SetBatchSize

Generic caller for all the transformations responsible to make model reshape-able by batch dimension.

#include <set_batch_size.hpp>

class SetBatchSize: public ov::pass::ModelPass
    // methods

    OPENVINO_RTTI("SetBatchSize", "0");
    virtual bool run_on_model(const std::shared_ptr<ov::Model>& m);

Inherited Members

    // typedefs

    typedef DiscreteTypeInfo type_info_t;

    // methods

    bool get_property(const PassPropertyMask& prop_mask) const;
    void set_name(const std::string& name);
    std::string get_name() const;
    void set_callback(const param_callback& callback);
    virtual void set_pass_config(const std::shared_ptr<PassConfig>& pass_config);
    std::shared_ptr<PassConfig> get_pass_config();
    bool transformation_callback(const std::shared_ptr<const Node>& node);
    virtual const type_info_t& get_type_info() const = 0;
    virtual bool run_on_model(const std::shared_ptr<ov::Model>& m) = 0;