Update header files #1

Rjasuja · Nov 20, 2023 · ebf4f2a · ebf4f2a
1 parent 9355739
commit ebf4f2a
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/tensorflow/lite/delegates/openvino/modular/graph_builder.h b/tensorflow/lite/delegates/openvino/modular/graph_builder.h
@@ -9,6 +9,7 @@ class GraphBuilder {
   void setResultNode(const int index, std::shared_ptr<ov::Node> resultNode);
   void setOutputAtOperandIndex(int index, ov::Output<ov::Node> output);
   ov::Output<ov::Node> getOperationOutput(int index);
+  void GraphBuilder(std::vector<int> tensors_to_replace);
 private:
   std::vector<ov::Output<ov::Node>> outputAtOperandIndex;
   std::vector<std::shared_ptr<ov::opset3::Parameter>> inputParams;

diff --git a/tensorflow/lite/delegates/openvino/modular/graph_manager.h b/tensorflow/lite/delegates/openvino/modular/graph_manager.h
@@ -1,5 +1,7 @@
-#include "tensorflow/lite/c/common."
+#include "tensorflow/lite/c/common.h"
 #include "tensorflow/lite/tools/logging.h"
+#include "graph_builder.h"
+#include "operation_builder.h"
 
 namespace tflite {
 namespace openvinodelegate {
@@ -8,13 +10,15 @@ class GraphManager {
   TfLiteStatus initializeGraph();
   TfLiteStatus generateGraph();
   void createInputParams();
+  void GraphManager(TfLiteContext* context, const TfLiteDelegateParams* params) : params_(params), context_(context) {}
 private:
   GraphBuilder* graphBuilder;
   OperationBuilder* opBuilder;
-  void instantitateOpClassFactory();
-  std::vector<std::shared_ptr<OperationsBase>> opNodes;
-  TfLiteContent* context;
-  const TfLiteDelegateParams* params;
+  //to instantiate classes for each op before actually calling createNode() for that op
+  void instantiateOpClassFactory();
+  std::vector<std::shared_ptr<OperationBuilder>> opNodes;
+  TfLiteContent* context_;
+  const TfLiteDelegateParams* params_;
   std::vector<int> tensors_to_replace;
 };
 

diff --git a/tensorflow/lite/delegates/openvino/modular/openvino_delegate_kernel.h b/tensorflow/lite/delegates/openvino/modular/openvino_delegate_kernel.h
@@ -20,6 +20,7 @@ limitations under the License.
 
 #include "NgraphNodes.h"
 #include "ov_utils.h"
+#include "ie_manager.h"
 #include "tensorflow/lite/builtin_ops.h"
 #include "tensorflow/lite/delegates/utils/simple_delegate.h"
 #include "tensorflow/lite/tools/logging.h"
@@ -35,6 +36,9 @@ class OpenVINODelegateKernel : public SimpleDelegateKernelInterface {
   TfLiteStatus Prepare(TfLiteContext *context, TfLiteNode *node) override;
 
   TfLiteStatus Eval(TfLiteContext *context, TfLiteNode *node) override;
+private:
+  GraphManager* graphManager;
+  IEManager* ieManager;
 };
 
 } // namespace openvinodelegate