Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.autograd.function._ContextMethodMixin Class Reference
Inheritance diagram for torch.autograd.function._ContextMethodMixin:
torch.autograd.function.BackwardCFunction torch.autograd.function.Function torch.autograd.function.InplaceFunction torch.autograd.function.NestedIOFunction torch.nn._functions.thnn.normalization.CrossMapLRN2d torch.nn._functions.thnn.sparse.EmbeddingBag torch.nn.modules._functions.SyncBatchNorm

Public Member Functions

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

 to_save
 
 dirty_tensors
 
 non_differentiable
 

Detailed Description

Definition at line 10 of file function.py.


The documentation for this class was generated from the following file: