raw_memory_object.h

Go to the documentation of this file.

#pragma once
 
#include "computation_graph/raw_memory_metadata.h"
#include "library_defines.h" // for EXPORT_OR_IMPORT
#include "ml_core/types.h"
#include <cstdint> // for int64_t
#include <vector>  // for vector
namespace blace {
 
class EXPORT_OR_IMPORT RawMemoryObject {
public:
  RawMemoryObject();
 
  ~RawMemoryObject();
 
  RawMemoryObject(void *data_ptr, ml_core::DataTypeEnum type,
                  ml_core::ColorFormatEnum color_format,
                  std::vector<int64_t> memory_sizes, ml_core::OrderEnum order,
                  ml_core::ValueRangeEnum value_range,
                  std::optional<blace::ml_core::BlaceHash> hash = std::nullopt,
                  bool copy_memory = true);
 
  RawMemoryObject(void *data_ptr, blace::RawMemoryMetadata meta_data,
                  bool copy_memory);
 
  RawMemoryObject(const RawMemoryObject &other);
 
  RawMemoryObject &operator=(const RawMemoryObject &other);
 
  RawMemoryObject(RawMemoryObject &&other);
 
  RawMemoryObject &operator=(RawMemoryObject &&other);
 
  void *get_data_ptr() const;
 
  std::vector<int64_t> get_memory_sizes() const;
 
  int get_memory_size() const;
 
  ml_core::DataTypeEnum get_type();
 
  ml_core::ColorFormatEnum get_color_format();
 
  ml_core::ValueRangeEnum get_value_range();
 
  ml_core::OrderEnum get_order();
 
  ml_core::BlaceHash get_hash() const;
 
  bool has_ownership();
 
  RawMemoryMetadata get_meta_data() const;
 
  template <class Archive>
  void serialize(Archive &ar, const unsigned int version) {}
 
private:
  void *_data_ptr;
 
  RawMemoryMetadata _memory_metadata;
 
  bool _take_memory_ownership = false;
};
 
} // namespace blace