GetRequiredInputTensor
Function Usage
Obtains the pointer to a required input tensor based on the input index in the operator prototype definition.
Prototype
const Tensor *GetRequiredInputTensor(const size_t ir_index) const
Parameters
|
Parameter |
Input/Output |
Description |
|---|---|---|
|
ir_index |
Input |
Index of the required input in the operator IR prototype definition, starting from 0. |
Returns
Pointer to a specified input tensor. If the input ir_index is invalid, a null pointer is returned.
For details about the definition of the tensor type, see Tensor.
Constraints
The tensor data address can only be obtained during the setting of data dependency. If the input is not set to data dependency, when this API is called to obtain a tensor, only the correct shape, format, and data type information can be obtained from the tensor, and the actual tensor data address (the obtained address is nullptr) cannot be obtained.
Examples
ge::graphStatus InferShape4ConcatD(TilingContext* context) {
auto in_tensor = context->GetRequiredInputTensor(0U);
...
}