@@ -359,8 +359,8 @@ def attribute(m):
359
359
360
360
361
361
#############################
362
- #
363
- # The following code will wrap PyTorch functions to check if channels last
362
+ # To debug which operators don't support channels last format, you could uncomment
363
+ # the following code. This will wrap PyTorch functions to check if channels last
364
364
# format is preserved through operations.
365
365
#
366
366
# attribute(torch.Tensor)
@@ -369,14 +369,14 @@ def attribute(m):
369
369
370
370
######################################################################
371
371
# If you found an operator that doesn't support channels last tensors
372
- # and you want to contribute, feel free to use following developers
373
- # guide https://github.com/pytorch/pytorch/wiki/Writing-memory-format-aware-operators.
372
+ # and you want to contribute, see following developers
373
+ # `Writing Memory Format Aware Operators < https://github.com/pytorch/pytorch/wiki/Writing-memory-format-aware-operators>`__ .
374
374
#
375
375
376
376
######################################################################
377
- # Code below is to recover the attributes of torch.
377
+ # The code below is to recover the attributes of torch.
378
378
#
379
- #for (m, attrs) in old_attrs.items():
379
+ # for (m, attrs) in old_attrs.items():
380
380
# for (k, v) in attrs.items():
381
381
# setattr(m, k, v)
382
382
0 commit comments