namespace ov::pass::transpose_sinking::utils::sink_backward¶
Overview¶
namespace sink_backward {
// global functions
ov::NodeVector InsertTransposeBeforeNode(
const std::shared_ptr<ov::Node>& main_node,
const std::shared_ptr<ov::opset10::Constant>& transpose_const,
std::vector<size_t> input_indexes = {},
std::function<std::shared_ptr<ov::Node>(const ov::Output<ov::Node>&node, size_t n_dims)> InsertUnsqueeze = InsertBroadcastUnsqueeze
);
} // namespace sink_backward
Detailed Documentation¶
Global Functions¶
ov::NodeVector InsertTransposeBeforeNode(
const std::shared_ptr<ov::Node>& main_node,
const std::shared_ptr<ov::opset10::Constant>& transpose_const,
std::vector<size_t> input_indexes = {},
std::function<std::shared_ptr<ov::Node>(const ov::Output<ov::Node>&node, size_t n_dims)> InsertUnsqueeze = InsertBroadcastUnsqueeze
)
Inserts transposes on inputs of.
main_node specified by
input_indexes with the order specified in
transpose_const. If
input_indexes is empty, then it inserts transposes for all inputs.