blace.ai
Loading...
Searching...
No Matches
blace::ops::FromRawMemoryFetcherOp Class Reference

#include <public_ops.h>

Inheritance diagram for blace::ops::FromRawMemoryFetcherOp:

Public Member Functions

 FromRawMemoryFetcherOp (std::shared_ptr< blace::RawMemoryFetcher > mem_fetcher, ml_core::DeviceEnum device)
 

Detailed Description

Operator to construct a node from custom memory.

Constructor & Destructor Documentation

◆ FromRawMemoryFetcherOp()

blace::ops::FromRawMemoryFetcherOp::FromRawMemoryFetcherOp ( std::shared_ptr< blace::RawMemoryFetcher >  mem_fetcher,
ml_core::DeviceEnum  device 
)

Constructs an operator based on custom memory (self or externally managed). The custom memory holds all metadata like sizes, datatype etc. to allow for full tracking of metadata inside the graph.

Parameters
mem_fetcherA shared_ptr of the memory fetcher.
deviceA enum indicating in which device the memory is stored.