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

Use ov namespace

Browse Source 
Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev committed 2021-12-08 14:19:15 +03:00
1 parent 568b151d1a
commit 64f14c73eb
2 files changed
+6 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

inference-engine/src/mkldnn_plugin/nodes/mkldnn_detection_output_node.cpp
+4 -4
View File
@@ -31,9 +31,9 @@ bool SortScorePairDescend<std::pair<int, int>>(const std::pair<float, std::pair<
} // namespace
bool MKLDNNDetectionOutputNode::isSupportedOperation(const std::shared_ptr<const ngraph::Node>& op, std::string& errorMessage) noexcept {
bool MKLDNNDetectionOutputNode::isSupportedOperation(const std::shared_ptr<const ov::Node>& op, std::string& errorMessage) noexcept {
try {
const auto doOp = ngraph::as_type_ptr<const ngraph::op::v8::DetectionOutput>(op);
const auto doOp = ov::as_type_ptr<const ov::op::v8::DetectionOutput>(op);
if (!doOp) {
errorMessage = "Node is not an instance of the DetectionOutput from the operations set v8.";
return false;
@@ -57,7 +57,7 @@ void MKLDNNDetectionOutputNode::createPrimitive() {
}
}
MKLDNNDetectionOutputNode::MKLDNNDetectionOutputNode(const std::shared_ptr<ngraph::Node>& op, const mkldnn::engine& eng,
MKLDNNDetectionOutputNode::MKLDNNDetectionOutputNode(const std::shared_ptr<ov::Node>& op, const mkldnn::engine& eng,
MKLDNNWeightsSharing::Ptr &cache) : MKLDNNNode(op, eng, cache) {
std::string errorMessage;
if (!isSupportedOperation(op, errorMessage)) {
@@ -72,7 +72,7 @@ MKLDNNDetectionOutputNode::MKLDNNDetectionOutputNode(const std::shared_ptr<ngrap
if (getOriginalOutputsNumber() != 1)
IE_THROW() << errorPrefix << "has incorrect number of output edges.";
auto doOp = ngraph::as_type_ptr<const ngraph::op::v8::DetectionOutput>(op);
auto doOp = ov::as_type_ptr<const ov::op::v8::DetectionOutput>(op);
auto attributes = doOp->get_attrs();
backgroundClassId = attributes.background_label_id;
inference-engine/src/mkldnn_plugin/nodes/mkldnn_detection_output_node.h
+2 -2
View File
@@ -12,7 +12,7 @@ namespace MKLDNNPlugin {
class MKLDNNDetectionOutputNode : public MKLDNNNode {
public:
MKLDNNDetectionOutputNode(const std::shared_ptr<ngraph::Node>& op, const mkldnn::engine& eng, MKLDNNWeightsSharing::Ptr &cache);
MKLDNNDetectionOutputNode(const std::shared_ptr<ov::Node>& op, const mkldnn::engine& eng, MKLDNNWeightsSharing::Ptr &cache);
void getSupportedDescriptors() override {};
void initSupportedPrimitiveDescriptors() override;
@@ -20,7 +20,7 @@ public:
void execute(mkldnn::stream strm) override;
bool created() const override;
static bool isSupportedOperation(const std::shared_ptr<const ngraph::Node>& op, std::string& errorMessage) noexcept;
static bool isSupportedOperation(const std::shared_ptr<const ov::Node>& op, std::string& errorMessage) noexcept;
protected:
void prepareParams() override;