MCPcopy Create free account
hub / github.com/NVIDIA/DALI / pytorch_case

Function pytorch_case

dali/test/python/test_dltensor_operator.py:165–166  ·  view source on GitHub ↗
(fun, device)

Source from the content-addressed store, hash-verified

163
164
165def pytorch_case(fun, device):
166 common_case(pytorch_wrapper(fun), device, partial(pytorch_compare, fun))
167
168
169def simple_pytorch_op(in1, in2):

Callers

nothing calls this directly

Calls 2

common_caseFunction · 0.85
pytorch_wrapperFunction · 0.85

Tested by

no test coverage detected