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

[GPU] MVN 1d dynamic batch case fix (#16826)

Browse Source
This commit is contained in:
Roman Lyamin
2023-04-11 09:42:51 +04:00
committed by GitHub
parent efc647a512
commit 234fe92931
2 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/plugins/intel_gpu/src/plugin/ops/parameter.cpp
View File

@@ -28,10 +28,19 @@ static void CreateParameterOp(Program& p, const std::shared_ptr<ngraph::op::v0::
auto inputInfo = networkInputs.at(op->get_friendly_name());
// first create and add the input layout
const auto inputDesc = inputInfo->getTensorDesc();
auto input_pshape = op->get_partial_shape();
InferenceEngine::Layout l = inputDesc.getLayout();
InferenceEngine::Precision ip = inputDesc.getPrecision();
auto input_pshape = op->get_partial_shape();
if (!p.use_new_shape_infer()) {
if (input_pshape.size() < 4) {
input_pshape.insert(input_pshape.end(), 4 - input_pshape.size(), ov::Dimension(1));
}
if (p.m_max_batch > 1) {
input_pshape[0] = ov::Dimension(p.m_curBatch);
}
}
cldnn::format inputFormat = cldnn::format::get_default_format(input_pshape.size());
std::vector<size_t> default_order(input_pshape.size());
std::iota(default_order.begin(), default_order.end(), 0);