Function trtweight

models/api.py:10–12 · view source on GitHub ↗

(weights: np.ndarray)

Source from the content-addressed store, hash-verified

8
9
10	def trtweight(weights: np.ndarray) -> trt.Weights:
11	weights = weights.astype(weights.dtype.name)
12	return trt.Weights(weights)
13
14
15	def get_width(x: int, gw: float, divisor: int = 8) -> int:

Conv2dFunction · 0.85

ConvFunction · 0.85

no outgoing calls

no test coverage detected