MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT-LLM / get_layer_by_name

Method get_layer_by_name

tensorrt_llm/network.py:403–405  ·  view source on GitHub ↗
(self, name: str)

Source from the content-addressed store, hash-verified

401 yield layer
402
403 def get_layer_by_name(self, name: str) -> Optional["Layer"]:
404 state = self._get_graph()
405 return state.name_to_layer.get(name, None)
406
407 def get_tensor_users(self, tensor) -> Iterable["Layer"]:
408 '''

Callers 3

removed_layersMethod · 0.95
build_engineFunction · 0.80

Calls 2

_get_graphMethod · 0.95
getMethod · 0.45

Tested by 1