blace.ai
|
#include <public_ops.h>
Public Member Functions | |
FromRawMemoryFetcherOp (std::shared_ptr< blace::RawMemoryFetcher > mem_fetcher, ml_core::DeviceEnum device) | |
Operator to construct a node from custom memory.
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.
mem_fetcher | A shared_ptr of the memory fetcher. |
device | A enum indicating in which device the memory is stored. |