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

Function dims_array

tensorrt_llm/_utils.py:93–99  ·  view source on GitHub ↗
(x)

Source from the content-addressed store, hash-verified

91
92
93def dims_array(x):
94 is_int64_dims = True
95 try:
96 trt.Dims([np.iinfo(np.int64).max])
97 except TypeError:
98 is_int64_dims = False
99 return int64_array(x) if is_int64_dims else int32_array(x)
100
101
102def bf16_array(x):

Callers 3

expand_dims_likeFunction · 0.85
splitFunction · 0.85
allgatherFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected