MutableOutputTdInfo
Description
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
CompileTimeTensorDesc *MutableOutputTdInfo(const size_t index) const
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
index |
Input |
Operator output index, starting from 0. |
Returns
Tensor description. The returned object is not const.
Restrictions
None
Example
for (size_t i = 0; i < node->GetAllOutDataAnchorsSize(); ++i) {
auto td = compute_node_info.MutableOutputTdInfo(i);
...
}
Parent topic: ComputeNodeInfo