graph_evaluator.h

Go to the documentation of this file.

#pragma once
#include "computation_graph/raw_memory_object.h"
#include "library_defines.h"
#include "ml_core/callbacks.h"
#include "public_ops.h"
#ifdef BLACE_AI_TORCH_INTERFACE
#include <ATen/core/TensorBody.h>
#endif
#include <filesystem>
#include <memory>
#ifdef BLACE_AI_OPENCV_INTERFACE
#include <opencv2/core/mat.hpp>
#endif
 
namespace blace {
class NodeOp;
}
 
namespace blace {
namespace computation_graph {
struct GraphEvaluatorImpl;
 
class EXPORT_OR_IMPORT GraphEvaluator {
public:
  GraphEvaluator(blace::ops::OpP node);
 
  GraphEvaluator(std::shared_ptr<NodeOp> node);
 
  void to_dot_file(const std::filesystem::path &filename);
 
  std::string to_dot_string();
 
#ifdef BLACE_AI_OPENCV_INTERFACE
  std::optional<cv::Mat> evaluateToCVMat();
#endif
 
  std::optional<std::string> evaluateToString();
 
  std::optional<std::vector<int64_t>> evaluateToIntList();
 
  std::optional<RawMemoryObject> evaluateToRawMemory(
      std::shared_ptr<ml_core::ProgressCallback> progress_callback = nullptr);
 
#ifdef BLACE_AI_TORCH_INTERFACE
  std::optional<at::Tensor> evaluateToTorchTensor(
      std::shared_ptr<ml_core::ProgressCallback> progress_callback = nullptr);
#endif
 
  ~GraphEvaluator();
 
private:
  std::unique_ptr<GraphEvaluatorImpl> _impl;
};
} // namespace computation_graph
} // namespace blace