#[repr(C)]pub struct LiteIO {
pub name: *const c_char,
pub is_host: c_int,
pub io_type: LiteIOType,
pub config_layout: LiteLayout,
}Expand description
\brief config the network input and output item
Fields§
§name: *const c_char! the tensor name in the graph corresponding to the IO
is_host: c_int! Used to mark where the input tensor comes from and the output where copy ! to, if is_host is true, the input is from host and output copy to host, ! otherwise device. Sometimes The input is from device and output no need ! copy to host, default is true.
io_type: LiteIOType! The IO type, it can be SHAPE or VALUE, when SHAPE is set, the input or ! output tensor value is invaid, only shape will be set, default is VALUE
config_layout: LiteLayout! The layout of the config from user, if other layout is set before ! forward or get after forward, this layout will by pass. if no other ! layout is set before forward, this layout will work. if this layout is ! no set, the model will forward with its origin layout. if in output, it ! will used to check.