openvino/model-optimizer/mo/ops
Vladimir Gavrilov 3aed19a46a
XLNET models bugs (#1199)
* Started to write equality comparator for StridedSlice layers.

* Now considered only unique StridedSlice consumers of non-constant data nodes.

* Fixed test for the transformation ConvertGroupedStridedSlice.

* Deleted commented code.

* Small fixes.

* Moved functions unique_by and group_by_with_binary_predicate to mo/utils/utils.py.

* Deleted function collect_sequences.

* Added asserts into the constructor of StridedSlice.

* Written test for the case when 1) there are 4 StridedSlice operations; 2) 2 of StridedSlice have the same data; 3) 2 others StridedSlice have the same data; 4) all StridedSlice operations outputs are consumed by different operations.

* Added some comments.
2020-11-19 10:07:30 +03:00
..
__init__.py Publishing R3 2018-10-16 13:45:03 +03:00
activation.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
assign.py publish master branch snapshot, revision ea98a886d925eb152931aab13856e68037665562 2020-05-22 03:42:00 +03:00
broadcast.py [MO] Fix: add PermuteInputs to shape broadcasting (#2419) 2020-10-05 12:48:57 +03:00
broadcast_test.py [MO] Add explicit broadcasting mode (#2077) 2020-09-25 13:26:47 +03:00
clamp.py [MO] Fix ONNX Clamp-11 shape infer with no min/max inputs (#2603) 2020-10-12 09:55:45 +03:00
concat.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
concat_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
const.py MO clean from IR v7 and other legacy code (#1521) 2020-07-29 17:43:12 +03:00
constant_of_shape.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
convolution.py [MO] Fix many spelling and grammar errors (#2543) 2020-10-09 12:16:12 +03:00
convolution_test.py added check to avoid IR generation in case of wrong input shape (#2127) 2020-09-16 11:29:05 +03:00
crop.py [MO] Fix when Crop asks for MXNet specific cmdarg `enable_ssd_gluoncv` (#1978) 2020-08-31 18:06:13 +03:00
crop_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
deconvolution.py [ MO ] Complete weights layout permutation (#2299) 2020-09-18 14:42:16 +03:00
deformable_convolution.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
eltwise.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
eltwise_n.py ONNX Loop operation support (#2756) 2020-10-27 23:04:43 +03:00
eltwise_ninputs_in_1.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
expand_dims.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
expand_dims_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
fill.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
flatten.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
group_norm.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
log_softmax.py Add LogSoftmax-5 to MO and ngraph (#2409) 2020-10-20 13:40:06 +03:00
lrn.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
lstmnonlinearity.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
memory.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
memoryoffset.py [MO] [Kaldi] Add TDNN Component (#1870) 2020-10-12 14:10:27 +03:00
op.py MO clean from IR v7 and other legacy code (#1521) 2020-07-29 17:43:12 +03:00
pad.py [ BTS & STB ] Fixing broken transformations (#874) 2020-07-03 19:55:00 +03:00
pad_test.py Publishing 2020.2 content 2020-04-13 21:17:23 +03:00
permute.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
pooling.py added check to avoid IR generation in case of wrong input shape (#2127) 2020-09-16 11:29:05 +03:00
pooling_test.py added check to avoid IR generation in case of wrong input shape (#2127) 2020-09-16 11:29:05 +03:00
power.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
read_value.py publish master branch snapshot, revision ea98a886d925eb152931aab13856e68037665562 2020-05-22 03:42:00 +03:00
reshape.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
result.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
roipooling.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
scale_shift.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
shape.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
slice.py ONNX Loop operation support (#2756) 2020-10-27 23:04:43 +03:00
slice_test.py ONNX Loop operation support (#2756) 2020-10-27 23:04:43 +03:00
softmax.py Add LogSoftmax-5 to MO and ngraph (#2409) 2020-10-20 13:40:06 +03:00
space_to_batch.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
squeeze.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
squeeze_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
strided_slice.py XLNET models bugs (#1199) 2020-11-19 10:07:30 +03:00
strided_slice_test.py Cleanup IR v7 from the MO (#1008) 2020-06-22 11:52:00 +03:00
tdnncomponent.py [MO] [Kaldi] Add TDNN Component (#1870) 2020-10-12 14:10:27 +03:00
tile.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
tile_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
unsqueeze.py Added transformation config to support automl efficientdet models (#2894) 2020-11-02 19:21:05 +03:00
unsqueeze_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00