Function torch2pb

python_backend/model.py:19–21 · view source on GitHub ↗

(name, tensor)

Source from the content-addressed store, hash-verified

17
18
19	def torch2pb(name, tensor):
20	return pb_utils.Tensor(name, tensor.numpy())
21	# return pb_utils.Tensor.from_dlpack(name, to_dlpack(tensor))
22
23
24	class TritonPythonModel:

executeMethod · 0.85

no outgoing calls

no test coverage detected