Implement LookupTableInsert shape inference (#2348)
authorRoman Kazantsev <roman.kazantsev@intel.com>
Tue, 20 Oct 2020 06:57:55 +0000 (09:57 +0300)
committerGitHub <noreply@github.com>
Tue, 20 Oct 2020 06:57:55 +0000 (09:57 +0300)
commitc2394508c158e46b9d2bff61eda5f94d0e4178d9
tree5fa822dc26fcce5afdca65aaae4ae580482313c4
parent347e92cc821dcbd1be2a6a71aa27961c77583edf
Implement LookupTableInsert shape inference (#2348)

* Implement LookupTableInsertV2 shape inference

It is needed if other nodes not beeing pruned in the graph
have a conditional dependence on LookupTableInsertV2 node.

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
* Fix after core-review #1

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
* Fix the code after review #2

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
* Fix after code review #3
model-optimizer/automation/package_BOM.txt
model-optimizer/extensions/front/tf/LookupTableInsert_ext.py [new file with mode: 0644]
model-optimizer/extensions/ops/LookupTableInsert.py [new file with mode: 0644]
model-optimizer/extensions/ops/LookupTableInsert_test.py [new file with mode: 0644]