MCPcopy
hub / github.com/NVIDIA/TensorRT-LLM / from_parameter

Function from_parameter

tensorrt_llm/layers/moe.py:184–187  ·  view source on GitHub ↗
(x)

Source from the content-addressed store, hash-verified

182 output_dtype = str_dtype_to_trt(output_dtype)
183
184 def from_parameter(x):
185 if isinstance(x, Parameter):
186 return x.value
187 return x
188
189 expert_weights_1 = from_parameter(expert_weights_1)
190 expert_weights_2 = from_parameter(expert_weights_2)

Callers 1

_moe_pluginFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected