Searched defs:output_process_fn (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/ |
| D | test_legacy_vmap.py | 2555 self, op, args, kwargs=None, output_process_fn=lambda x: x, batch_size=3 argument
2586 self, op, args, kwargs=None, output_process_fn=lambda x: x, batch_size=3 argument
|
| /external/pytorch/test/functorch/ |
| D | test_ops.py | 680 output_process_fn, argument
|
| D | test_vmap.py | 3621 self, op, args, kwargs=None, output_process_fn=lambda x: x, batch_size=3 argument
3655 self, op, args, kwargs=None, output_process_fn=lambda x: x, batch_size=3 argument
|