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);