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

Function shape

tensorrt_llm/_torch/auto_deploy/utils/node_utils.py:933–936  ·  view source on GitHub ↗
(node: Node)

Source from the content-addressed store, hash-verified

931
932
933def shape(node: Node) -> Tuple[int, ...]:
934 if not has_shape(node):
935 return None
936 return node.meta["val"].shape
937
938
939def get_weight_tensor(gm: GraphModule, node: Node) -> "torch.Tensor":

Callers 15

get_weight_shapeFunction · 0.70
get_layer_after_linear_nodeFunction · 0.70
validateMethod · 0.50
_get_dim0_from_argFunction · 0.50
_insert_sharded_mxfp4_mlp_epFunction · 0.50
_determine_fused_weight_dimsFunction · 0.50
detect_sharding_from_configFunction · 0.50
detect_column_row_shardFunction · 0.50
detect_dp_bmm_shardFunction · 0.50
transpose_for_scoresMethod · 0.50
forwardMethod · 0.50
merge_cachesMethod · 0.50

Calls 1

has_shapeFunction · 0.85

Tested by

no test coverage detected