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 <memory>
#include <vector> // for vector
namespace blace {
class RawMemoryObjectImpl;
}
 
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,
                  ml_core::DeviceEnum device,
                  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) noexcept;
 
  RawMemoryObject &operator=(RawMemoryObject &&other) noexcept;
 
  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::DeviceEnum get_device();
 
  ml_core::BlaceHash get_hash() const;
 
  bool has_ownership();
 
  RawMemoryMetadata get_meta_data() const;
 
  bool is_similar(RawMemoryObject obj);
 
  blace::RawMemoryObjectImpl *get_impl();
 
private:
  std::unique_ptr<blace::RawMemoryObjectImpl> impl_;
};
 
} // namespace blace