Skip to content

Commit

Permalink
generic node enabled (#9)
Browse files Browse the repository at this point in the history 
* generic node enabled

* fixes after review
  • Loading branch information
Maxim Andronov committed Jul 20, 2021
1 parent 4f30522 commit a04fba3
Show file tree
Hide file tree
Showing 10 changed files with 85 additions and 382 deletions.
179 changes: 3 additions & 176 deletions inference-engine/src/mkldnn_plugin/CMakeLists.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,182 +16,9 @@ if (ENABLE_CPU_DEBUG_CAPS)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DCPU_DEBUG_CAPS")
endif()

file(GLOB SOURCES
${CMAKE_CURRENT_SOURCE_DIR}/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/mkldnn/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/utils/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/utils/rt_info/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/common/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/emitters/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/ngraph_transformations/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/ngraph_transformations/op/*.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_split_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reorder_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_input_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_eltwise_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_convert_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reference_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_conv_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_fake_quantize_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_tensoriterator_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_batch_to_space_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_broadcast_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_tile_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_bucketize_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_concat_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_rnn.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_bin_conv_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_ctc_greedy_decoder_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_ctc_greedy_decoder_seq_len_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_ctc_loss_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_select_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_transpose_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_topk_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_strided_slice_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_space_to_depth_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_shuffle_channels_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_scatter_update_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_roll_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_roi_pooling_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_roi_align_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_psroi_pooling_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reverse_sequence_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reshape_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reorg_yolo_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_region_yolo_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reduce_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_range_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_pad_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_one_hot_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_normalize_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_non_max_suppression_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_mvn_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_cum_sum_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_deconv_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_memory_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_math_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_lrn_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_log_softmax_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_interpolate_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_detection_output_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_generate_proposals_single_image_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_priorgridgenerator_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_roifeatureextractor_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_topkrois_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_bag_offset_sum_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_bag_packed_sum_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_bag_sum_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_segments_sum_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_space_to_batch_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_grn_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_tree_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_nd_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_elements_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_extract_image_patches_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_dft_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_detection_output_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_depth_to_space_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_proposal_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_softmax_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_fullyconnected_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_matmul_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_pooling_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_def_conv_node.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/proposal_imp.cpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/list.cpp
)

file(GLOB HEADERS
${CMAKE_CURRENT_SOURCE_DIR}/*.h
${CMAKE_CURRENT_SOURCE_DIR}/*.hpp
${CMAKE_CURRENT_SOURCE_DIR}/mkldnn/*.h
${CMAKE_CURRENT_SOURCE_DIR}/mkldnn/*.hpp
${CMAKE_CURRENT_SOURCE_DIR}/utils/*.h
${CMAKE_CURRENT_SOURCE_DIR}/utils/rt_info/*.hpp
# ${CMAKE_CURRENT_SOURCE_DIR}/nodes/*.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_split_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reorder_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_input_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_eltwise_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_convert_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reference_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_conv_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_fake_quantize_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_tensoriterator_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_batch_to_space_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_concat_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_rnn.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_bin_conv_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_broadcast_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_tile_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_bucketize_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_ctc_greedy_decoder_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_ctc_greedy_decoder_seq_len_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_ctc_loss_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_select_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_transpose_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_topk_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_strided_slice_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_space_to_depth_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_shuffle_channels_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_scatter_update_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_roll_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_roi_pooling_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_roi_align_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_psroi_pooling_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reverse_sequence_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reshape_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reorg_yolo_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_region_yolo_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_reduce_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_range_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_pad_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_one_hot_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_normalize_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_non_max_suppression_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_mvn_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_cum_sum_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_deconv_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_memory_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_math_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_lrn_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_log_softmax_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_interpolate_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_detection_output_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_generate_proposals_single_image_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_priorgridgenerator_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_roifeatureextractor_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_experimental_detectron_topkrois_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_bag_offset_sum_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_bag_packed_sum_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_bag_sum_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_embedding_segments_sum_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_space_to_batch_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_grn_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_tree_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_nd_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_gather_elements_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_extract_image_patches_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_dft_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_detection_output_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_depth_to_space_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_proposal_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_softmax_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_fullyconnected_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_matmul_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_pooling_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/mkldnn_def_conv_node.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/proposal_imp.hpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/base.hpp
${CMAKE_CURRENT_SOURCE_DIR}/nodes/common/*.h
${CMAKE_CURRENT_SOURCE_DIR}/nodes/common/*.hpp
)

#file(GLOB_RECURSE SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp)
#file(GLOB_RECURSE HEADERS ${CMAKE_CURRENT_SOURCE_DIR}/*.h
# ${CMAKE_CURRENT_SOURCE_DIR}/*.hpp)
file(GLOB_RECURSE SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp)
file(GLOB_RECURSE HEADERS ${CMAKE_CURRENT_SOURCE_DIR}/*.h
${CMAKE_CURRENT_SOURCE_DIR}/*.hpp)

addVersionDefines(mkldnn_plugin.cpp CI_BUILD_NUMBER)

Expand Down
11 changes: 11 additions & 0 deletions inference-engine/src/mkldnn_plugin/cpu_memory_desc_utils.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,11 @@
#include "cpu_memory_desc_utils.h"
#include "mkldnn_memory.h"
#include "utils/general_utils.h"
#include "utils/cpu_utils.hpp"
#include <limits>
#include <vector>
#include <numeric>
#include <blob_factory.hpp>

using namespace mkldnn;
using namespace MKLDNNPlugin;
Expand Down Expand Up @@ -380,4 +382,13 @@ MemoryDescPtr MemoryDescUtils::resetOffset(const MemoryDesc* desc) {
return desc->clone();
}

InferenceEngine::Blob::Ptr MemoryDescUtils::interpretAsBlob(const MKLDNNMemory &mem) {
// TODO [DS]: Rewrite when IE is moved to the new TensorDescriptor
auto& memDesc = mem.GetDesc();
InferenceEngine::TensorDesc desc = convertToTensorDesc(memDesc);

desc = InferenceEngine::TensorDesc(desc.getPrecision(), memDesc.getShape().getStaticDims(), desc.getBlockingDesc());
return MKLDNNPlugin::isEmptyTensorDesc(desc) ? make_blob_with_precision(desc) : make_blob_with_precision(desc, mem.GetData());
}

} // namespace MKLDNNPlugin
2 changes: 2 additions & 0 deletions inference-engine/src/mkldnn_plugin/cpu_memory_desc_utils.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
namespace MKLDNNPlugin {
class MKLDNNMemoryDesc;
class BlockedMemoryDesc;
class MKLDNNMemory;

class MemoryDescUtils {
public:
Expand All @@ -26,6 +27,7 @@ class MemoryDescUtils {
static MemoryDescPtr applyUndefinedOffset(const MKLDNNMemoryDesc& desc);
static MemoryDescPtr applyUndefinedOffset(const BlockedMemoryDesc& desc);
static MemoryDescPtr resetOffset(const MemoryDesc* desc);
static InferenceEngine::Blob::Ptr interpretAsBlob(const MKLDNNMemory& mem);
};

} // namespace MKLDNNPlugin
13 changes: 2 additions & 11 deletions inference-engine/src/mkldnn_plugin/mkldnn_graph.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -947,13 +947,13 @@ Config MKLDNNGraph::getProperty() const {

void MKLDNNGraph::getInputBlobs(InferenceEngine::BlobMap &resp) {
for (auto &it : inputNodesMap) {
resp[it.first] = convertMemToBlob(it.second->getChildEdgeAt(0)->getMemory());
resp[it.first] = MemoryDescUtils::interpretAsBlob(it.second->getChildEdgeAt(0)->getMemory());
}
}

void MKLDNNGraph::getOutputBlobs(InferenceEngine::BlobMap &resp) {
for (auto &it : outputNodesMap) {
resp[it.first] = convertMemToBlob(it.second->getParentEdgeAt(0)->getMemory());
resp[it.first] = MemoryDescUtils::interpretAsBlob(it.second->getParentEdgeAt(0)->getMemory());
}
}

Expand Down Expand Up @@ -1188,12 +1188,3 @@ void MKLDNNGraph::EnforceBF16() {
InferenceEngine::CNNNetwork MKLDNNGraph::dump() const {
return dump_graph_as_ie_ngraph_net(*this);
}

InferenceEngine::Blob::Ptr MKLDNNGraph::convertMemToBlob(const MKLDNNMemory &mem) const {
// TODO [DS]: Rewrite when IE is moved to the new TensorDescriptor
auto& memDesc = mem.GetDesc();
InferenceEngine::TensorDesc desc = MemoryDescUtils::convertToTensorDesc(memDesc);

desc = InferenceEngine::TensorDesc(desc.getPrecision(), memDesc.getShape().getStaticDims(), desc.getBlockingDesc());
return isEmptyTensorDesc(desc) ? make_blob_with_precision(desc) : make_blob_with_precision(desc, mem.GetData());
}
1 change: 0 additions & 1 deletion inference-engine/src/mkldnn_plugin/mkldnn_graph.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -226,7 +226,6 @@ class MKLDNNGraph {

private:
void EnforceBF16();
InferenceEngine::Blob::Ptr convertMemToBlob(const MKLDNNMemory& mem) const;
};

} // namespace MKLDNNPlugin
27 changes: 13 additions & 14 deletions inference-engine/src/mkldnn_plugin/mkldnn_node.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1190,20 +1190,19 @@ MKLDNNNode* MKLDNNNode::NodesFactory::create(const std::shared_ptr<ngraph::Node>
const MKLDNNExtensionManager::Ptr& extMgr, MKLDNNWeightsSharing::Ptr &w_cache) {
MKLDNNNode *newNode = nullptr;
std::string errorMessage;
// TODO [DS]: uncomment
// try {
// std::unique_ptr<MKLDNNNode> ol(createNodeIfRegistered(MKLDNNPlugin, Generic, op, eng, w_cache));
// if (ol != nullptr && ol->created(extMgr))
// newNode = ol.release();
// } catch (const InferenceEngine::Exception& ex) {
// IE_SUPPRESS_DEPRECATED_START
// if (ex.getStatus() != NOT_IMPLEMENTED) {
// throw;
// } else {
// errorMessage += getExceptionDescWithoutStatus(ex);
// }
// IE_SUPPRESS_DEPRECATED_END
// }
try {
std::unique_ptr<MKLDNNNode> ol(createNodeIfRegistered(MKLDNNPlugin, Generic, op, eng, w_cache));
if (ol != nullptr && ol->created(extMgr))
newNode = ol.release();
} catch (const InferenceEngine::Exception& ex) {
IE_SUPPRESS_DEPRECATED_START
if (ex.getStatus() != NOT_IMPLEMENTED) {
throw;
} else {
errorMessage += getExceptionDescWithoutStatus(ex);
}
IE_SUPPRESS_DEPRECATED_END
}

if (newNode == nullptr) {
try {
Expand Down
Loading

0 comments on commit a04fba3

Please sign in to comment.