 |
Caffe2 - Python API
A deep learning, cross platform ML framework
|
|
Caffe2 - Python API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| def | backward (self, gradients) |
| def | forward (self, args) |
| def | save_for_backward (self, args) |
| def | saved_tensors (self) |
| def | mark_dirty (self, args, kwargs) |
| def | mark_non_differentiable (self, args, kwargs) |
| def | forward_extended (self, input) |
| def | backward_extended (self, grad_output) |
 Public Member Functions inherited from torch.autograd.function.FunctionMeta | |
| def | __init__ (cls, name, bases, attrs) |
| Public Member Functions inherited from torch.autograd.function._ContextMethodMixin | |
| def | save_for_backward (self, tensors) |
| def | mark_dirty (self, args) |
| def | mark_shared_storage (self, pairs) |
| def | mark_non_differentiable (self, args) |
Public Attributes | |
| retain_variables | |
| to_save | |
| dirty_tensors | |
| non_differentiable | |
| Public Attributes inherited from torch.autograd.function._ContextMethodMixin | |
| to_save | |
| dirty_tensors | |
| non_differentiable | |
Additional Inherited Members | |
| Static Public Member Functions inherited from torch.autograd.function.Function | |
| def | forward (ctx, args, kwargs) |
| def | backward (ctx, grad_outputs) |
| Static Public Attributes inherited from torch.autograd.function.Function | |
| is_traceable | |
Definition at line 332 of file function.py.
1.8.11
