Searched defs:transformation_fn (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/ |
| D | test_torch.py | 5403 def _test_memory_format_transformations(self, device, input_generator_fn, transformation_fn, argument
5463 def transformation_fn(tensor, **kwargs): function
5480 def transformation_fn(tensor, **kwargs): function
5497 def transformation_fn(tensor, **kwargs): function
5542 def transformation_fn(tensor, **kwargs): function
|
| D | test_reductions.py | 1410 def _test_memory_format_transformations(self, device, input_generator_fn, transformation_fn, argument
|