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

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

Examples

ge::graphStatus InferShape4ConcatD(TilingContext* context) {
  auto in_tensor = context->GetRequiredInputTensor(0U);
  ...
}