export.py File Reference

Classes
class	export.OnnxWrapper
class	export.OnnxBackboneWrapper
class	export.OnnxExpertWrapper

Namespaces
namespace	export

Functions
Dict[str, torch.Tensor]	export._extract_state_dict (object ckpt_obj)
	export._write_u32s (f, values)
	export._write_f32_array (f, np.ndarray arr)
None	export.export_native_weights (Path out_path, ChessNetFactorizedMoE model)
None	export.export_inputs_and_refs (Path inputs_path, Path refs_path, ChessNetFactorizedMoE model, int n_samples, int seed)
None	export.main ()

Variables
int	export.MAGIC_WEIGHTS = 0x32454F4D
int	export.MAGIC_INPUTS = 0x32504E49
int	export.MAGIC_REFS = 0x32464552
int	export.VERSION = 1
int	export.MAX_BRANCH_PLANES = 10
dict	export.POOL_TO_CODE