class ov::pass::transpose_sinking::TSForwardBase¶
TSForwardBase is a base class for all forward transformations.
#include <ts_base.hpp>
class TSForwardBase: public ov::pass::MatcherPass
{
public:
// typedefs
typedef std::function<bool(const std::shared_ptr<Node>&main_node, const utils::TransposeInputsInfo&transpose_info)> sinking_function;
// methods
OPENVINO_RTTI("ov::pass::TSForwardBase", "0");
template <class... Types>
void create_pattern(
bool const_transpose_input,
std::vector<size_t> transpose_indices = {}
);
void transpose_sinking(
const std::string& pass_name,
const sinking_function& sinking_transformation = nullptr
);
};
// direct descendants
class TSBinaryForward;
class TSConcatForward;
class TSCumSumForward;
class TSDataMovementForward;
class TSGatherForward;
class TSInterpolateForward;
class TSReductionForward;
class TSShapeOfForward;
class TSSliceForward;
class TSSplitForward;
class TSSqueezeForward;
class TSTileForward;
class TSUnaryForward;
class TSUnsqueezeForward;
Inherited Members¶
public:
// 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;
OPENVINO_RTTI("ov::pass::MatcherPass");
MatcherPass& operator = (const MatcherPass&);
bool apply(std::shared_ptr<ov::Node> node);
template <typename T, class... Args>
std::shared_ptr<T> register_new_node(Args&&... args);
template <typename T>
std::shared_ptr<T> register_new_node(const std::shared_ptr<T>& node);
std::shared_ptr<ov::Node> register_new_node_(const std::shared_ptr<ov::Node>& node);
const std::vector<std::shared_ptr<ov::Node>>& get_new_nodes();
void clear_new_nodes();
std::shared_ptr<pattern::Matcher> get_matcher();