Searched refs:wrap_fn (Results 1 – 2 of 2) sorted by relevance
790 def wrap_fn(func): function793 return wrap_fn
1277 wrap_fn = (1281 enqueue_ops.append(wrap_fn(device=host_device, op_fn=enqueue_ops_fn))1317 wrap_fn = (1322 wrap_fn(device=host_device, op_fn=enqueue_ops_fn))