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


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 ()



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 \

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