Visible to Intel only — GUID: GUID-2F98D59D-4C31-453C-85E1-4D932C4C3D01
Visible to Intel only — GUID: GUID-2F98D59D-4C31-453C-85E1-4D932C4C3D01
class dnnl::graph::compiled_partition
Overview
A compiled partition object. More…
#include <dnnl_graph.hpp>
class compiled_partition: public compiled_partition_handle
{
public:
// construction
compiled_partition();
compiled_partition(dnnl_graph_compiled_partition_t compiled_partition);
// methods
logical_tensor query_logical_tensor(size_t tid) const;
std::vector<std::pair<size_t, size_t>> get_inplace_ports() const;
void execute(
stream& astream,
const std::vector<tensor>& inputs,
const std::vector<tensor>& outputs
) const;
};
Detailed Documentation
A compiled partition object.
Construction
compiled_partition()
Default constructor. Constructs an empty object.
compiled_partition(dnnl_graph_compiled_partition_t compiled_partition)
Constructs a compiled partition object.
Methods
logical_tensor query_logical_tensor(size_t tid) const
Queries an input or output logical tensor according to tensor ID.
If the tensor ID doesn’t belong to any input or output of the compiled partition, an exception will be raised by the API.
Parameters:
tid |
The unique id of required tensor. |
Returns:
The logical tensor.
std::vector<std::pair<size_t, size_t>> get_inplace_ports() const
Returns the hint of in-place pairs from a compiled partition.
It indicates that an input and an output of the partition can share the same memory buffer for computation. In-place computation helps to reduce the memory footprint and improves cache locality. But since the library may not have a global view of user’s application, it’s possible that the input tensor is used at other places in user’s computation graph. In this case, the user should take the in-place pair as a hint and pass a different memory buffer for output tensor to avoid overwriting the input memory buffer which will probably cause unexpected incorrect results.
Returns:
A list of pairs of input and output IDs.
void execute(
stream& astream,
const std::vector<tensor>& inputs,
const std::vector<tensor>& outputs
) const
Execute a compiled partition.
Parameters:
astream |
Stream object to run over. |
inputs |
A list of input tensors. |
outputs |
A list of output tensors. |