Skip to main content

set_dev_input

Module.set_dev_input(name , tensor : Tensor)

Sets the input data to the pre-allocated memory with the given tensor name and input data.

Note

  • If the original tensor data in host memory is not stored in a contiguous layout, the tensor data will be copied automatically to contiguous host CPU memory.

  • Before calling this function, you must pre-allocate the tensor parameter by creating a Tensor object.

  • For multi-device models, the tensor must be on COMPND device.

  • The pre-allocated tensor must match the model's output tensor in format, data type, and shape, as retrieved via tcim_lite.runtime.Module.get_output_info(). This ensures that the output data can be correctly stored.

  • This function should be called before tcim_lite.runtime.Module.run().

Parameters:

  • name (str) -- The name of the input tensor.

  • tensor (Tensor) -- The input Tensor object.