MutableInputTdInfo

Description

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

CompileTimeTensorDesc *MutableInputTdInfo(const size_t index) const

Parameters

Returns

Tensor description. The returned object is not const.

Restrictions

None

Example

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