Xp/revise embedding bag (#7694)
* Fix: variable writing error.
mas_dimension -> max_dimension
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Update description.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Update EmbeddingBagOffsetsSum comment, it will be easy to understand how to calculate bag.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Add attribute description for md
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* WIP: Add correctness test.
Known issue:
1: weight fp16 issue.
2: default idx(1), don't choose in segment id. why?
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* change param order.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Support test T_IND type (int32 and int64)
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* weight support multiple type input.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* refactor correctness test code.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* embeddingbag offset sum correctness test pass.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* embeddingbag packed sum correctness test pass.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Add visitor without attribute test.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Add EmbeddingSegmentsSum serialization test.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Add EmbeddingBagPackedSum, EmbeddingBagOffsetsSum serialization test.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* fix build error after rebase.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* unify name: from ngraph:: to ov::
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* add fp64 support
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Remove my MACRO define wrappers.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Remove useless header file.
* tmp. pass.
* Remove MACRO warpper for embeddingbag_offsetsum
* Remove myown wrapper header, and update template plugin.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Change vector to ov::Shape.
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Rename CreateConstantVV-> CreateConstant
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>
* Fix rebase issue. duplicate case ov::element::f64:
Signed-off-by: Yan, Xiping <xiping.yan@intel.com>