namespace ov::op::util::embedding

Overview

namespace embedding {

// global functions

template <class TShape>
TShape out_shape_infer(
    const ov::Node \* op,
    const TShape& emb_table_shape,
    const TShape& dim_shape_src
    );

} // namespace embedding

Detailed Documentation

Global Functions

template <class TShape>
TShape out_shape_infer(
    const ov::Node \* op,
    const TShape& emb_table_shape,
    const TShape& dim_shape_src
    )

Return a copy of the emb_table_shape with the first dimension replaced by the first dimension from the dim_shape_src

Parameters:

TShape	Shape type
op	Pointer to operator.
emb_table_shape	The shape to be copied
dim_shape_src	The shape to copy the first dimension from, with dynamic or static rank > 1

Returns:

The copy of the emb_table_shape with the first dimsnsion overwritten by dim_shape_src[0] if the rank is static, otherwise fully dynamic shape with dynamic rank.