get_inputs - 2.0 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2022-01-20
Version
2.0 English
Gets all input TensorBuffers of RunnerExt.

Prototype

List[vart.TensorBuffer] get_inputs()

Parameters

None.

Returns

All input TensorBuffers. A vector of raw pointer to the input TensorBuffer.

Usage

input_tensor_buffers = runner.get_inputs()
input_dim = tuple(input_tensor_buffers[0].get_tensor().dims)