MutableInputTdInfo

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

CompileTimeTensorDesc *MutableInputTdInfo(const size_t index) const

Parameters

Returns

Tensor description. The returned object is not const.

Constraints

None

Examples

1
2
3
4

  for (size_t i = 0; i < compute_node_info.GetInputsNum(); ++i) {
    auto td = compute_node_info.MutableInputTdInfo(i);
    ...
  }