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

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

Examples

ge::graphStatus Tiling4ReduceCommon(TilingContext* context) {
  auto in_shape = context->GetInputShape(0);
  GE_ASSERT_NOTNULL(in_shape);
  auto axes_tensor = context->GetOptionalInputTensor(1);
  ...
}