InputTensorDesc
Function Usage
Sets the description of Tensor. It is used to construct information (like ComputeNodeInfo) in the base class ExtendedKernelContext of TilingParseContext.
Prototype
1 | OpTilingParseContextBuilder &InputTensorDesc(size_t index, ge::DataType dtype, ge::Format origin_format, ge::Format storage_format, const gert::ExpandDimsType &expand_dims_type = {}) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
index |
Input |
Index of the operator input instance. |
dtype |
Input |
Data type of the input Tensor. |
origin_format |
Input |
Original format of the input Tensor. |
storage_format |
Input |
Storage format of the input Tensor. |
expand_dims_type |
Input |
ExpandDimsType of a dimension expansion rule for the input Tensor. The default value is {}. |
Returns
A reference to the OpTilingParseContextBuilder object. It is used for chaining method calls.
Constraints
Before the Build method call, InputTensorDesc must be called to set Tensor description information. Otherwise, the constructed TilingParserContext will contain undefined data.