GetOptionalInputTensor

Function Usage

Obtains the pointer to an optional input tensor based on the input index in the operator prototype definition.

Prototype

const Tensor *GetOptionalInputTensor(const size_t ir_index) const

Parameters

Parameter	Input/Output	Description
ir_index	Input	Index of the optional input in the operator IR prototype definition, starting from 0.

Returns

Pointer to the input tensor of a specified ir_index. If the input ir_index is invalid or the input is not instantiated, 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 Tiling4ReduceCommon(TilingContext* context) {
  auto in_shape = context->GetInputShape(0);
  GE_ASSERT_NOTNULL(in_shape);
  auto axes_tensor = context->GetOptionalInputTensor(1);
  ...
}

Parent topic: TilingContext