Implement support for opset3 EmbeddingBag ops (#546) * [MO] Implement EmbeddingBag_3 * Transform dynamic sub-graph of Wide and Deep into EmbeddingSegmentsSum - Expressed SparseWeightedSum sub-graph through EmbeddingSegmentsSum - Removed experimental SparseWeightedSum layer - Implemented tests for the transformation Signed-off-by: Roman Kazantsev * Fix EmbeddingBag shape infer * Fix EmbeddingSegmentsSum transformation for Wide and Deep Signed-off-by: Roman Kazantsev * Fix EmbeddingSegmentSum replacer after ports swap Signed-off-by: Roman Kazantsev * Update package_BOM.txt Signed-off-by: Roman Kazantsev * Add unit tests for EmbeddingXXX shape infer * Fix ATen resolver * Remove deleted files from BOM * Add opset version to embedding_bag * Use base class for EmbeddingBag * Fix per_sample_weights case * Fix EmbeddingSegmentsSum transformation Signed-off-by: Roman Kazantsev * Fix EmbeddingBag checks * Fix ATen front transformation and merge conflicts * Fix BOM * Work around limitation for I64 input of W&D model Signed-off-by: Roman Kazantsev * Cleanup where operation to fix affect of WhereDecomposition transform Signed-off-by: Roman Kazantsev * Fix BOM * Correct EmbeddingSegmentSum transform for Wide and Deep Add casting segment ids to i32 and remove ConstToResult sub-graph. Signed-off-by: Roman Kazantsev * Update BOM with RemoveConstToResult transform Signed-off-by: Roman Kazantsev * Add more comments for RemoveConstToResult transformation Signed-off-by: Roman Kazantsev * Remove useless logging in EmbeddingSegmentsSum transformation Signed-off-by: Roman Kazantsev * Small fixes * Move EmbeddingBag resolving back to front phase * Improve error messages * Fix typo in unittests * Reimplement sparse_reshape middle transform Avoid deprecated API. Signed-off-by: Roman Kazantsev * Clean-up graph after sparse_reshape and ConstToResult transformation Signed-off-by: Roman Kazantsev * Fix clean-up for transformations Signed-off-by: Roman Kazantsev * Fix clean-up for transformation #2 Signed-off-by: Roman Kazantsev Co-authored-by: Roman Kazantsev

最好能把所有东西打包成docker #2

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions