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.