GetOptionalInputDesc

Description

Obtains the tensor description of an optional input based on the input index in the operator prototype definition.

Prototype

const CompileTimeTensorDesc *GetOptionalInputDesc(const size_t ir_index) const;

Parameters

Returns

Pointer to the input TensorDesc. If the input index is invalid or the input is not instantiated, a null pointer is returned.

For details about the definition of CompileTimeTensorDesc, see CompileTimeTensorDesc.

Restrictions

None

Example

// Assume that KernelContext *context already exists.
auto extend_context = reinterpret_cast<ExtendedKernelContext *>(context);
// Case 1: Assume that input 0 of the IR prototype of an operator is optional and there is actually one input.
auto optional_input_td = extend_context->GetOptionalInputDesc(0); // Obtain the tensor description of optional input 0.
// Case 2: Assume that input 0 of the IR prototype of an operator is optional and there is actually no input.
auto optional_input_td = extend_context->GetOptionalInputDesc(0); // nullptr