kernels-test
/

backward-marker-test

Model card Files Files and versions

danieldk HF Staff commited on Apr 8

Commit

f60b4b0

·

1 Parent(s): 2913ead

Import

Files changed (1) hide show

torch-ext/kernels_test/layers.py +1 -0

torch-ext/kernels_test/layers.py CHANGED Viewed

@@ -1,5 +1,6 @@
 import torch
 import torch.nn as nn
 class LinearImplicitBackward(nn.Module):

 import torch
 import torch.nn as nn
+from torch.nn import functional as F
 class LinearImplicitBackward(nn.Module):