GetOutputTdInfo
Function Usage
Obtains the output tensor description of an operator. Note that the shape information that cannot be determined during compilation is not included in the tensor description.
Prototype
1 | const CompileTimeTensorDesc *GetOutputTdInfo(const size_t index) const |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
index |
Input |
Output index of an operator, starting from 0. |
Returns
Tensor description of the const type
Constraints
None
Examples
1 2 | auto compute_node_info = extend_kernel_context->GetComputeNodeInfo(); auto output_td = compute_node_info->GetOutputTdInfo(output_index); |
Parent topic: ComputeNodeInfo