GetComputeNodeInfo

Description

Obtains information about the compute node corresponding to the kernel.

Graph execution, in essence, is the execution of the kernel of each node in the graph. ComputeNodeInfo contains information such as InputDesc, and this method exactly can be used to obtain the saved ComputeNodeInfo from KernelContext.

Prototype

const ComputeNodeInfo *GetComputeNodeInfo() const

Parameters

None

Returns

Information about the compute node.

For details about the definition of ComputeNodeInfo, see ComputeNodeInfo.

Restrictions

None

Example

// Assume that KernelContext *context already exists.
auto extend_context = reinterpret_cast<ExtendedKernelContext *>(context);
auto compute_node_info = extend_context->GetComputeNodeInfo();