OpenVINO™ is an open-source toolkit for optimizing and deploying AI inference
Go to file
Xiping Yan 3354275da1
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>
2021-11-15 13:49:21 +03:00
.ci Enable GPU in static build w/o oneDNN (#8563) 2021-11-15 13:44:11 +03:00
.github [Python API] Move samples and docs to the new directory (#7851) 2021-10-14 14:49:35 +03:00
cmake Enable GPU in static build w/o oneDNN (#8563) 2021-11-15 13:44:11 +03:00
docs Xp/revise embedding bag (#7694) 2021-11-15 13:49:21 +03:00
inference-engine Xp/revise embedding bag (#7694) 2021-11-15 13:49:21 +03:00
licensing Update third party files (#8382) 2021-11-03 15:29:42 +03:00
model-optimizer Check if argv has compress_fp16 attribute (#8545) 2021-11-12 22:54:10 +03:00
ngraph Xp/revise embedding bag (#7694) 2021-11-15 13:49:21 +03:00
openvino Enable GPU in static build w/o oneDNN (#8563) 2021-11-15 13:44:11 +03:00
runtime Fixed ie_wheel build (#8565) 2021-11-12 23:05:53 +03:00
samples samples: Print verbose error messages to stderr (#7795) 2021-11-10 11:42:52 +03:00
scripts [IE Sample Scripts] Use cmake to build samples (#8442) 2021-11-10 17:31:28 +03:00
tests [TF FE] Implement and refactor tensorflow layer tests (#8051) 2021-11-12 11:03:45 +03:00
thirdparty Enable GPU in static build w/o oneDNN (#8563) 2021-11-15 13:44:11 +03:00
tools [POT] Add pattern se blocks (#8425) 2021-11-15 12:53:07 +03:00
.gitattributes [POT] Update tests with new data (#8209) 2021-10-27 12:40:19 +03:00
.gitignore [POT] Added missed file to POT (#8118) 2021-10-21 11:28:26 +03:00
.gitmodules Moved Post-training Optimization Tool to open-source (#7940) 2021-10-15 16:35:35 +03:00
CMakeLists.txt Added support of external modules in static build (#8518) 2021-11-12 08:56:57 +03:00
CODEOWNERS CODEOWNERS: Add /tools/pot/ @openvinotoolkit/openvino-pot-maintainers 2021-11-02 13:55:02 +03:00
install_build_dependencies.sh Added reporting of unresolved symbols for plugins (#7810) 2021-10-05 04:26:01 +03:00
Jenkinsfile Beautify Jenkinsfile a little bit 2021-05-31 15:24:56 +03:00
LICENSE Publishing R3 2018-10-16 13:45:03 +03:00
README.md [Docs] Add PyPI downloads badge (#7610) 2021-09-23 09:30:57 +00:00
SECURITY.md Added SECURITY.md back (#3177) 2020-11-17 16:44:44 +03:00

OpenVINO™ Toolkit

Stable release Apache License Version 2.0 GitHub branch checks state Azure DevOps builds (branch) PyPI Downloads

This toolkit allows developers to deploy pre-trained deep learning models through a high-level C++ Inference Engine API integrated with application logic.

This open source version includes several components: namely Model Optimizer, nGraph and Inference Engine, as well as CPU, GPU, MYRIAD, multi device and heterogeneous plugins to accelerate deep learning inferencing on Intel® CPUs and Intel® Processor Graphics. It supports pre-trained models from the Open Model Zoo, along with 100+ open source and public models in popular formats such as Caffe*, TensorFlow*, MXNet* and ONNX*.

Repository components:

License

Deep Learning Deployment Toolkit is licensed under Apache License Version 2.0. By contributing to the project, you agree to the license and copyright terms therein and release your contribution under these terms.

Resources:

Support

Please report questions, issues and suggestions using:


* Other names and brands may be claimed as the property of others.