|
def | forward (self, input, weight, bias, running_mean, running_var, eps, momentum, process_group, world_size) |
|
def | backward (self, grad_output) |
|
def | forward (ctx, args, kwargs) |
|
def | backward (ctx, grad_outputs) |
|
|
| process_group |
|
| world_size |
|
| to_save |
|
| dirty_tensors |
|
| non_differentiable |
|
|
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) |
|
| is_traceable |
|
Definition at line 5 of file _functions.py.
The documentation for this class was generated from the following file: