GetInputDesc
Description
Obtains the input TensorDesc of the operator based on its input name or input index.
Prototype
APIs using strings will be deprecated in later versions. Use APIs supporting non-string data instead.
TensorDesc GetInputDesc(const std::string &name) const
TensorDesc GetInputDescByName(const char_t *name) const
TensorDesc GetInputDesc(uint32_t index) const
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
name |
Input |
Input name of the operator. If no operator input name is available, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format). |
index |
Input |
Input index of the operator. If no operator input index is available, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format). |
Returns
TensorDesc of the operator input
Exception Handling
None
Restrictions
None