Skip to content

Commit

Permalink
[NPU] - Port plugin changes for dynamic shapes support - code style
Browse files Browse the repository at this point in the history
  • Loading branch information
DariaMityagina committed Apr 2, 2024
1 parent 121b06a commit 4150d85
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,8 @@ ZeroInferRequest::ZeroInferRequest(const std::shared_ptr<ZeroInitStructsHolder>&
const std::string shapeBufferName = SHAPE_TENSOR_PREFIX + inputName;
const IONodeDescriptor& shapeDescriptor = _metadata.shapes.at(inputName);

check_level_zero_attributes_match(shapeDescriptor, executorInputDescriptors.at(shapeBufferName),
check_level_zero_attributes_match(shapeDescriptor,
executorInputDescriptors.at(shapeBufferName),
shapeBufferName);
allocate_tensor(inputName, shapeDescriptor, !STATE_TENSOR, allocator, SHAPE_TENSOR);
}
Expand All @@ -138,7 +139,8 @@ ZeroInferRequest::ZeroInferRequest(const std::shared_ptr<ZeroInitStructsHolder>&
const std::string shapeBufferName = SHAPE_TENSOR_PREFIX + outputName;
const IONodeDescriptor& shapeDescriptor = _metadata.shapes.at(outputName);

check_level_zero_attributes_match(shapeDescriptor, executorOutputDescriptors.at(shapeBufferName),
check_level_zero_attributes_match(shapeDescriptor,
executorOutputDescriptors.at(shapeBufferName),
shapeBufferName);
allocate_tensor(outputName, shapeDescriptor, !STATE_TENSOR, allocator, SHAPE_TENSOR);
}
Expand Down

0 comments on commit 4150d85

Please sign in to comment.