namespace ov::pass::transpose_sinking::utils::sink_forward

Overview

namespace sink_forward {

// global functions

bool UpdateInputTransposes(
    const std::shared_ptr<ov::Node>& main_node,
    const TransposeInputsInfo& transpose_input_info,
    std::vector<size_t> input_indexes = {}
    );

void RemoveInputNode(const std::shared_ptr<ov::Node>&, size_t input_idx);

ov::NodeVector InsertOutputTransposes(
    const std::shared_ptr<ov::Node>& main_node,
    const TransposeInputsInfo& transpose_input_info
    );

} // namespace sink_forward

Detailed Documentation

Global Functions

bool UpdateInputTransposes(
    const std::shared_ptr<ov::Node>& main_node,
    const TransposeInputsInfo& transpose_input_info,
    std::vector<size_t> input_indexes = {}
    )

Inserts reversed transposed on @args main_node inputs. Removes input transpose specified in.

• transpose_input_info

void RemoveInputNode(const std::shared_ptr<ov::Node>&, size_t input_idx)

Removes.

• input node

ov::NodeVector InsertOutputTransposes(
    const std::shared_ptr<ov::Node>& main_node,
    const TransposeInputsInfo& transpose_input_info
    )

Inserts transposes on each main_node output with the order specified in.

• transpose_input_info