IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE CLDNN] StridedSliceTransformation removed from GPU plugin (#4016)

Browse Source
This commit is contained in:
Vladislav Golubev 2021-01-26 18:44:29 +03:00 committed by GitHub
parent 247606baf6
commit 9371393311
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
inference-engine/src/cldnn_engine/cldnn_engine.cpp
View File

@ -62,6 +62,7 @@
#include <low_precision/transformer.hpp>
#include <low_precision/mat_mul.hpp>
#include <low_precision/strided_slice.hpp>
#include "cldnn_engine.h"
#include "cldnn_executable_network.h"
@ -286,7 +287,9 @@ InferenceEngine::CNNNetwork clDNNEngine::CloneAndTransformNetwork(const Inferenc
LayerTransformation::QuantizedTensorAlignment::None, // quantizedTensorAlignmentOnWeights
true); // supportAsymmetricQuantization
LowPrecisionTransformer transformer(LowPrecisionTransformer::getAllTransformations(params)
.add<MatMulTransformation, ngraph::opset1::MatMul>(LayerTransformation::Params(params).setSupportAsymmetricQuantization(false)));
.add<MatMulTransformation, ngraph::opset1::MatMul>(LayerTransformation::Params(params).setSupportAsymmetricQuantization(false))
// INT8 StridedSlice not supported
.remove<StridedSliceTransformation, ngraph::opset1::StridedSlice>());
transformer.transform(nGraphFunc);
}

2
inference-engine/tests/functional/plugin/gpu/shared_tests_instances/low_precision_transformations/strided_slice_transformation.cpp
View File

@ -86,7 +86,7 @@ const std::vector<LayerTestsDefinitions::StridedSliceTransformationParam> params
}
};
INSTANTIATE_TEST_CASE_P(DISABLED_smoke_LPT, StridedSliceTransformation,
INSTANTIATE_TEST_CASE_P(smoke_LPT, StridedSliceTransformation,
::testing::Combine(
::testing::ValuesIn(netPrecisions),
::testing::Values(ngraph::Shape({ 1, 3, 24, 24 })),

3
inference-engine/tests/functional/plugin/gpu/shared_tests_instances/skip_tests_config.cpp
View File

@ -49,5 +49,8 @@ std::vector<std::string> disabledTestPatterns() {
R"(.*(smoke_DetectionOutput3In).*)",
R"(.*(smoke_DetectionOutput5In).*)",
R"(.*(ScatterUpdateLayerTest).*)",
// INT8 StridedSlice not supported
R"(.*(LPT/StridedSliceTransformation).*)",
};
}