Function pytorch_wrapper

dali/test/python/test_dltensor_operator.py:135–136 · view source on GitHub ↗

(fun)

Source from the content-addressed store, hash-verified

133
134
135	def pytorch_wrapper(fun):
136	return lambda in1, in2: pytorch_adapter(fun, in1, in2)
137
138
139	def common_case(wrapped_fun, device, compare, synchronize=True):

pytorch_caseFunction · 0.85

pytorch_adapterFunction · 0.85

no test coverage detected