GetInputTdInfo

Function Usage

Obtains the input tensor description of an operator based on the input index information. Note that the shape information that cannot be determined during compilation is not included in the tensor description.

Prototype

1

const CompileTimeTensorDesc *GetInputTdInfo(const size_t index) const

Parameters

Returns

Tensor description of the const type

Constraints

None

Examples

1
2

auto compute_node_info = extend_kernel_context->GetComputeNodeInfo();
auto input_td = compute_node_info->GetInputTdInfo(input_index);