|
def | forward (cls, ctx, weight, indices, offsets, max_norm, norm_type, scale_grad_by_freq, mode) |
|
def | __init__ (cls, name, bases, attrs) |
|
def | save_for_backward (self, tensors) |
|
def | mark_dirty (self, args) |
|
def | mark_shared_storage (self, pairs) |
|
def | mark_non_differentiable (self, args) |
|
|
def | backward (ctx, grad_output) |
|
def | forward (ctx, args, kwargs) |
|
def | backward (ctx, grad_outputs) |
|
|
| to_save |
|
| dirty_tensors |
|
| non_differentiable |
|
| is_traceable |
|
Definition at line 13 of file sparse.py.
The documentation for this class was generated from the following file: