get_dev_input

Module.get_dev_input(name)

Gets the input tensor to the pre-allocated memory with the given tensor name.

Note

If the tensor name is set via tcim_lite.runtime.Option.set_dummy_tensors(), this function returns an error.

Parameters:

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

Returns:

tensor -- Returns the input tensor.

Return type:

Tensor

Note

This function does not automatically copy tensor data to contiguous host CPU memory. You need to call tcim_lite.runtime.Tensor.to_host() and tcim_lite.runtime.Tensor.cast_to() to explicitly copy and convert tensor data to host CPU memory if needed.