export.OnnxWrapper Class Reference

Inheritance diagram for export.OnnxWrapper:

Collaboration diagram for export.OnnxWrapper:

Public Member Functions
	__init__ (self, ChessNetFactorizedMoE model)
	forward (self, torch.Tensor branch_in, torch.Tensor bypass_in, torch.Tensor global_in)

Public Attributes
	model = model
	planes_per_type = list(model.PLANES_PER_TYPE)

Constructor & Destructor Documentation

__init__()

export.OnnxWrapper.__init__	(		self,
		ChessNetFactorizedMoE	model )

Here is the call graph for this function:

Here is the caller graph for this function:

Member Function Documentation

forward()

export.OnnxWrapper.forward	(		self,
		torch.Tensor	branch_in,
		torch.Tensor	bypass_in,
		torch.Tensor	global_in )

Member Data Documentation

model

export.OnnxWrapper.model = model

planes_per_type

export.OnnxWrapper.planes_per_type = list(model.PLANES_PER_TYPE)

---

The documentation for this class was generated from the following file:

• training/export.py