Parameter

Versioned name: Parameter-1

Category: Infrastructure

Short description: Parameter layer specifies input to the model.

Attributes:

  • element_type

  • Description: the type of element of output tensor

  • Range of values: u8, u16, u32, u64, i8, i16, i32, i64, f16, f32, boolean, bf16
  • Type: string
  • Default value: None

  • Required: Yes

  • shape

  • Description: the shape of the output tensor

  • Range of values: list of non-negative integers, empty list is allowed that means 0D or scalar tensor
  • Type: int[]
  • Default value: None
  • Required: Yes

Example

<layer ... type="Parameter" ...>
    <data>element_type="f32" shape="1,3,224,224"</data>
    <output>
        <port id="0">
            <dim>1</dim>
            <dim>3</dim>
            <dim>224</dim>
            <dim>224</dim>
        </port>
    </output>
</layer>