adding OCM git submodule

Author: ck-intel

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "ocm"]
+	path = ocm
+	url = https://github.com/intel/ocm

CMakeLists.txt

@@ -198,6 +198,7 @@ add_subdirectory(third-party)
 add_subdirectory(logging)
 add_subdirectory(ngraph_bridge)
 add_subdirectory(tools)
+add_subdirectory(ocm/OCM)
 
 # The following targets depend on the Tensorflow source code directory
 # Get the absolute file name for the source

ngraph_bridge/CMakeLists.txt

@@ -69,6 +69,7 @@ target_link_libraries(
         absl_container
         absl_strings
         ${InferenceEngine_LIBRARIES}
+        ocm
     )
 
 target_compile_definitions( 
@@ -109,6 +110,7 @@ endif()
 set(IE_LIBS ${OPENVINO_ARTIFACTS_DIR}/deployment_tools/inference_engine/lib/intel64/)
 set(TBB_LIBS ${OPENVINO_ARTIFACTS_DIR}/deployment_tools/inference_engine/external/tbb/lib/)
 install(DIRECTORY ${IE_LIBS} DESTINATION "${NGTF_INSTALL_LIB_DIR}" )
+install(FILES ${CMAKE_INSTALL_PREFIX}/../ocm/OCM/libocm.so DESTINATION "${NGTF_INSTALL_LIB_DIR}")
 install(DIRECTORY ${TBB_LIBS} DESTINATION "${NGTF_INSTALL_LIB_DIR}" )
 if(USE_OPENVINO_FROM_LOCATION)
     file(GLOB NGRAPH_LIB_FILES "${IE_LIBS}/lib*" "${TBB_LIBS}/lib*")

ngraph_bridge/ngraph_rewrite_pass.cc

@@ -31,6 +31,8 @@
 #include "ngraph_bridge/ngraph_mark_for_clustering.h"
 #include "ngraph_bridge/ngraph_utils.h"
 
+#include "ocm/include/ocm_nodes_checker.h"
+
 using namespace std;
 
 namespace tensorflow {
@@ -104,7 +106,30 @@ class NGraphEncapsulationPass : public NGraphRewritePass {
 
     // 1. Mark for clustering then, if requested, dump the graphs.
     std::set<string> skip_these_nodes = {};
-    TF_RETURN_IF_ERROR(MarkForClustering(graph, skip_these_nodes));
+    // TF_RETURN_IF_ERROR(MarkForClustering(graph, skip_these_nodes));
+
+    // OCM bypassing the MarkForClustering function call
+    const char* device_id =  std::getenv("NGRAPH_TF_BACKEND");
+    if (device_id==nullptr){
+      device_id = "CPU";
+    }
+    std::string ov_version = "2021_1";
+    ocm::Framework_Names fName = ocm::Framework_Names::TF;
+    ocm::FrameworkNodesChecker FC(fName, device_id, ov_version, options.graph->get());
+    std::vector<void *> nodes_list = FC.MarkSupportedNodes();
+
+    // cast back the nodes in the TF format and mark the nodes for clustering (moved out from MarkForClustering function)
+    const std::map<std::string, SetAttributesFunction>& set_attributes_map = GetAttributeSetters();
+    for (auto void_node : nodes_list) {
+    // TODO(amprocte): move attr name to a constant
+      tensorflow::Node* node = (tensorflow::Node *)void_node;
+      node->AddAttr("_ngraph_marked_for_clustering", true);
+      auto it = set_attributes_map.find(node->type_string());
+      if (it != set_attributes_map.end()) {
+        it->second(node);
+      }
+    }
+    
     util::DumpTFGraph(graph, idx, "marked");
 
     // 2. Assign clusters then, if requested, dump the graphs.

ocm

@@ -143,6 +143,7 @@ target_link_libraries(
     tensorflow_ops_testutil
     absl_synchronization
     ${InferenceEngine_LIBRARIES} ${TBB_IMPORTED_TARGETS}
+    ocm
 )
 
 # First install the libngraph_bridge.so and headers