List of Header Files

atb/types.h

Definitions of basic data structures and constants.

atb/svector.h

Definitions of vectors on the stack.

atb/context.h

Context API class, which is used to set the Operation context, such as the stream and tiling attributes.

atb/operation.h

Definitions of operator APIs.

atb/infer_op_params.h

Definitions of inference operator parameters.

atb/train_op_params.h

Definitions of training operator parameters.

atb/graph_op_builder.h

Definitions of a method for building a graph operator.

atb/utils.h

Definitions of tool APIs.

atb/atb_infer.h

Only references to the preceding header files are included. Reference to this header file can provide references to all the preceding files.

atb/atb_comm.h

API of the ATB communicator.