elasticai.creator.ir2torch.default_handlers#

Module Contents#

Functions#

linear

relu

conv1d

sigmoid

flatten

Data#

handlers

API#

elasticai.creator.ir2torch.default_handlers.handlers#

[]

elasticai.creator.ir2torch.default_handlers.linear(impl: elasticai.creator.torch2ir.Implementation) → torch.nn.Module[source]#
elasticai.creator.ir2torch.default_handlers.relu(impl: elasticai.creator.torch2ir.Implementation) → torch.nn.Module[source]#
elasticai.creator.ir2torch.default_handlers.conv1d(impl: elasticai.creator.torch2ir.Implementation) → torch.nn.Conv1d[source]#
elasticai.creator.ir2torch.default_handlers.sigmoid(impl: elasticai.creator.torch2ir.Implementation) → torch.nn.Sigmoid[source]#
elasticai.creator.ir2torch.default_handlers.flatten(imp: elasticai.creator.torch2ir.Implementation) → torch.nn.Flatten[source]#