Caffe2 - Python API
A deep learning, cross platform ML framework
Functions | Variables
tools.autograd.gen_autograd Namespace Reference

Functions

def format_return_type (returns)
 
def get_simple_type (arg)
 
def load_aten_declarations (path)
 
def load_deprecated_signatures (aten_decls, deprecated_path)
 
def gen_autograd (aten_path, out, autograd_dir)
 
def main ()
 

Variables

 VIEW_FUNCTIONS
 
 RETURNS_VIEWS_OF_INPUT
 

Detailed Description

To run this file by hand from the root of the PyTorch
repository, run:

python -m tools.autograd.gen_autograd \
       build/aten/src/ATen/Declarations.yaml \
       $OUTPUT_DIR

Where $OUTPUT_DIR is where you would like the files to be
generated.  In the full build system, OUTPUT_DIR is
torch/csrc/autograd/generated/