natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
openvino/model-optimizer/extensions/middle
History
Roman Kazantsev 12fb83dc1e
Correct layout for Einsum inputs and output (#6696) 
* Fix recovery of output subscript in Einsum implicit mode

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Fix code style

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Correct layout adjustment for Einsum inputs and output

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Correct a comment in the unit-test

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Setup correct transformation dependencies for LayoutChangeForEinsum

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
 		2021-07-23 11:01:06 +03:00
..
AddFakeQuantizeFuse.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
AddIsCyclicAttribute.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
AddMeanScaleValues.py fix incorrect input names for mean values (#5508) 2021-05-06 13:33:51 +03:00
ApplyNHWCtoNCHWpermutation.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ApplyPermutations.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ArgOpsToTopK.py Implementation of ArgMin ONNX + TF extractors (#5126) 2021-05-06 13:41:49 +03:00
AttributedTileNormalizer.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
BiasAddBroadcasting.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
BinarizeWeightsM1P1.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
BlockLSTMtoLSTMSequence.py [ MO ] Warnings fix (#5105) 2021-04-08 14:09:00 +03:00
CheckForCycle.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ConcatOptimization.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ConstSwitchResolver.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ConvToBinaryConv.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ConvertGroupedStridedSlice.py Provide `keep_output_port` attribute to `add_opoutput` function (#6117) 2021-07-05 12:34:44 +05:00
ConvertLayoutDependentOperations.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ConvertMultiInputConv.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
CustomSubgraphCall.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
CutInputHavingZeroDimFromConcat.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
DecomposeBias.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
DecomposeBidirectionalRNNSequence.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
Deconvolution3rdInputNormalization.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
DeleteControlFlowEdges.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
DeleteNotExecutable.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
DilatedConvolution.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
EltwiseChecker.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
EltwiseInputReshape.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
FakeSplitOutputs.py Add keep split output ports without consumers (#5136) 2021-04-12 17:49:53 +03:00
FuseReshapesSequence.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
FusedBatchNormNonConstant.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
FusedBatchNormTraining.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
GRURNNSequenceToTensorIterator.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
GatherNdNormalizer.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
GroupNorm.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
InputCut.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
InsertLayoutPropagationTransposes.py MO support for operations DFT and IDFT (#5197) 2021-05-07 09:44:24 +03:00
InsertSelect.py Kaldi priors (#6258) 2021-07-02 15:18:23 +05:00
InterpolateSequenceToInterpolate.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
L2NormFusing.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
LSTMRNNSequenceToTensorIterator.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
LayoutChangeForConstantShapePaths.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
LayoutChangeForEinsum.py Correct layout for Einsum inputs and output (#6696) 2021-07-23 11:01:06 +03:00
LeakyReluPattern.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
MXNetRNNSequenceNormalize.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
MXNetSplitMultiLayers.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
MXTileReplacer.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
MakeKaldiConstReshapable.py fix bug: assert is keyword, not function (#5131) 2021-04-08 12:02:18 +03:00
MarkSubgraphsWithCorrectLayout.py [ MO ] Warnings fix (#5105) 2021-04-08 14:09:00 +03:00
MoveConstToLoopBody.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
MulFakeQuantizeFuse.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ONNXRNNSequenceNormalize.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ONNXResize11ToInterpolate.py Add support resize with 2 inputs (#5927) 2021-06-08 13:15:54 +03:00
PartialInfer.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
PoolV2ToAttributedPool.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
RNNSequenceNormalizeToIE.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ReluQuantizeFuse.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
RemoveDuplicationMemory.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
RemoveIdentity.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
RemoveRedundantReshapeAfterCropAndResize.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
RemoveRedundantReshapes.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
RemoveUselessConcatSplit.py [ MO ] Warnings fix (#5105) 2021-04-08 14:09:00 +03:00
RemoveUselessCrops.py [ MO ] Warnings fix (#5105) 2021-04-08 14:09:00 +03:00
RemoveUselessPad.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ReplaceMemoryOffsetWithSplice.py Kaldi priors (#6258) 2021-07-02 15:18:23 +05:00
ReplacePNorm.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ReplaceSpliceNodePattern.py [ MO ] Warnings fix (#5105) 2021-04-08 14:09:00 +03:00
ReverseTransposeNormalization.py Fix transpose with reverse (#6234) 2021-06-24 18:12:32 +03:00
ReverseV2ToReverseSequence.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
ScaleInput.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
SequenceLengthToMask.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
SharedWeightsDuplication.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
SliceConverter.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
SliceLikeToStridedSlice.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
SplitConcatPairToInterpolate.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
StridedSliceNormalizer.py [ MO ] Warnings fix (#5105) 2021-04-08 14:09:00 +03:00
SwapAxesMiddleReplacer.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TF_lstm_cell_to_generic.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TensorIteratorBackEdge.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TensorIteratorCondition.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TensorIteratorConditionChecker.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TensorIteratorInput.py [ MO ] Warnings fix (#5105) 2021-04-08 14:09:00 +03:00
TensorIteratorLSTMToLSTMSequence.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TensorIteratorMerge.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TensorIteratorOutput.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
TensorIterator_utils.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
UnsqueezeTileReshapeBlockToInterpolate.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
UpsampleToResample.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
UselessMerge.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
UselessSplitEraser.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
__init__.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
dequantize_linear_resolver.py Add axis support (#5242) 2021-05-17 15:03:22 +03:00
fusings.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
pass_separator.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
permute_tensor_iterator.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
preprocessing.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
quantize_fuses.py Add axis support (#5242) 2021-05-17 15:03:22 +03:00
quantize_linear_resolver.py Add axis support (#5242) 2021-05-17 15:03:22 +03:00
reverse_tensor_iterator.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
sparse_reshape.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
split_tdnn_memoryoffset.py fix batch adding to init value of read value (#4187) 2021-03-31 11:32:36 +03:00