extract_module_info

class braintrace.extract_module_info(model, *model_args, **model_kwargs)

Extracting the model information for the etrace compiler.

Parameters:

• model (Module) – The model to extract the information.

• model_args – The arguments of the model.

• model_kwargs – The keyword arguments of the model.

Return type:

ModuleInfo

Returns:

The model information, instance of ModuleInfo.