cavis/contrib/codegen-tools/libnd4j-gen
agibsonccc 46dbd0b203 Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
..
src/main/java/org/nd4j/descriptor Update copyrights remove attic and relocate elsewhere 2021-02-09 13:16:31 +09:00
README.md Dev commits 2021-02-01 14:31:20 +09:00
op-ir.proto Dev commits 2021-02-01 14:31:20 +09:00
pom.xml Update LICENSE 2021-02-01 17:47:29 +09:00

README.md

Libnd4j Op Descriptor Generator

This module contains a few files for generating op descriptors for libnd4j. An op descriptor contains its number of inputs and outputs as well as the names in the code of those attributes when they are assigned.

The main class parses a root directory specified by the user containing the libnd4j code base. In the libnd4j code base, it will scan for op signatures and certain macros found in https://github.com/eclipse/deeplearning4j/blob/master/libnd4j/include/system/op_boilerplate.h

From that, it will automatically generate a set of op descriptors including counts of types of ops as well as names. The up to date OpDescriptor class can be found here The main class can be found here