Convolution configuration

class torch_lattice.nn.functional.conv.conv_config.Dataflow(*values)

Bases: Enum

ImplicitGEMM = 0

GatherScatter = 1

FetchOnDemand = 2

CodedCSR = 3

torch_lattice.nn.functional.conv.conv_config.keys_check(conv_config)

torch_lattice.nn.functional.conv.conv_config.get_global_conv_config()

torch_lattice.nn.functional.conv.conv_config.set_global_conv_config(conv_config)

torch_lattice.nn.functional.conv.conv_config.clear_global_conv_config()

torch_lattice.nn.functional.conv.conv_config.get_default_conv_config(conv_mode=ConvMode.mode0, training=False)

Parameters:

• conv_mode (ConvMode)

• training (bool)

torch_lattice.nn.functional.conv.conv_mode.get_kmap_mode()

torch_lattice.nn.functional.conv.conv_mode.set_kmap_mode(kmap_mode)

Parameters:

kmap_mode (str)

torch_lattice.nn.functional.conv.conv_mode.get_downsample_mode()

torch_lattice.nn.functional.conv.conv_mode.set_downsample_mode(downsample_mode)

Parameters:

downsample_mode (str)

class torch_lattice.nn.functional.conv.conv_mode.ConvMode(*values)

Bases: Enum

mode0 = 0

mode1 = 1

mode2 = 2

torch_lattice.nn.functional.conv.conv_mode.get_conv_mode()

torch_lattice.nn.functional.conv.conv_mode.set_conv_mode(conv_mode)