MCPcopy Index your code
hub / github.com/NVIDIA/TensorRT-LLM / get_inputs

Method get_inputs

tensorrt_llm/network.py:336–343  ·  view source on GitHub ↗

Get the inputs of the network. Returns: Iterable[Tensor]

(self)

Source from the content-addressed store, hash-verified

334 return profiles
335
336 def get_inputs(self):
337 '''
338 Get the inputs of the network.
339
340 Returns:
341 Iterable[Tensor]
342 '''
343 return self._inputs.values()
344
345 def get_outputs(self):
346 '''

Callers 9

to_dotMethod · 0.95
_get_network_hashMethod · 0.95
test_num_inputsMethod · 0.45
test_is_inputMethod · 0.45
test_get_tensor_usersMethod · 0.45
rewriteMethod · 0.45
get_input_keyMethod · 0.45
buildMethod · 0.45
replace_all_uses_withMethod · 0.45

Calls 1

valuesMethod · 0.45

Tested by 5

test_num_inputsMethod · 0.36
test_is_inputMethod · 0.36
test_get_tensor_usersMethod · 0.36
rewriteMethod · 0.36
get_input_keyMethod · 0.36