GetComputeNodeInfo

Function Usage

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 can be used to obtain the saved ComputeNodeInfo from KernelContext.

Prototype

1

const ComputeNodeInfo *GetComputeNodeInfo() const

Parameters

None

Returns

Information about the compute node.

For details about the definition of ComputeNodeInfo, see ComputeNodeInfo.

Constraints

None

Examples

1
2
3

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