get_inputs - 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)