Fixed transformations looking for FusedBatchNorm operation to look for FBNV2 and...
authorEvgeny Lazarev <evgeny.lazarev@intel.com>
Thu, 12 Nov 2020 04:33:39 +0000 (07:33 +0300)
committerGitHub <noreply@github.com>
Thu, 12 Nov 2020 04:33:39 +0000 (07:33 +0300)
commitf4d399f471d93366cf023bde939be73b4c293a93
treea9e4185fb1d0147b6f43d30fb71332a7c844f831
parent9420b6e5990be21aa107defc76b423c436a5da35
Fixed transformations looking for FusedBatchNorm operation to look for FBNV2 and FBNV3 also (#3078)

* Fixed transformations looking for FusedBatchNorm operation to consider FusedBatchNormV2 and FusedBatchNormV3 also.

* Updated unit test for FusedBatchNormTraining

* Fixed unit test
model-optimizer/extensions/front/tf/mvn.py
model-optimizer/extensions/middle/FusedBatchNormNonConstant.py
model-optimizer/extensions/middle/FusedBatchNormTraining.py
model-optimizer/extensions/middle/FusedBatchNormTraining_test.py