Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
class dnnl::graph::tensor
Overview
A tensor object. More…
#include <dnnl_graph.hpp>
class tensor: public tensor_handle
{
public:
// construction
tensor();
tensor(const logical_tensor& lt, const engine& aengine, void* handle);
tensor(const logical_tensor& lt, const engine& aengine);
// methods
void* get_data_handle() const;
void set_data_handle(void* handle);
engine get_engine() const;
logical_tensor get_logical_tensor() const;
};Detailed Documentation
A tensor object.
Construction
tensor()Default constructor. Constructs an empty object.
tensor(const logical_tensor& lt, const engine& aengine, void* handle)Constructs a tensor object according to a given logical tensor, an engine, and a memory handle.
Parameters:
lt |
The given logical tensor |
aengine |
Engine to store the data on. |
handle |
Handle of memory buffer to use as an underlying storage.
|
tensor(const logical_tensor& lt, const engine& aengine)Constructs a tensor object.
The underlying buffer for the memory will be allocated by the library.
Parameters:
lt |
The given logical tensor |
aengine |
Engine to store the data on. |
Methods
void* get_data_handle() constReturns the underlying memory buffer.
On the CPU engine, or when using USM, this is a pointer to the allocated memory.
void set_data_handle(void* handle)Sets the underlying memory handle.
Parameters:
handle |
Memory handle. |
engine get_engine() constReturns the associated engine.
Returns:
An engine object
logical_tensor get_logical_tensor() constReturns the logical tensor of a tensor object.
Returns:
A logical_tensor object.