xnn.common.models.dispersion_fast.atm.TripletPlan

class xnn.common.models.dispersion_fast.atm.TripletPlan(order, src, dst, launches, n_atoms)

Bases: object

The center-sorted edge list of one evaluation and its kernel launches.

Parameters:

• order (Tensor)

• src (Tensor)

• dst (Tensor)

• launches (list)

• n_atoms (int)