hls4ml.converters.pytorch package
Submodules
hls4ml.converters.pytorch.convolution module
- hls4ml.converters.pytorch.convolution.parse_conv1d_layer(pytorch_layer, layer_name, input_shapes, data_reader, config)
- hls4ml.converters.pytorch.convolution.parse_conv2d_layer(pytorch_layer, layer_name, input_shapes, data_reader, config)
hls4ml.converters.pytorch.core module
- hls4ml.converters.pytorch.core.parse_activation_layer(pytorch_layer, layer_name, input_shapes, data_reader, config)
- hls4ml.converters.pytorch.core.parse_batchnorm_layer(pytorch_layer, layer_name, input_shapes, data_reader, config)
- hls4ml.converters.pytorch.core.parse_linear_layer(pytorch_layer, layer_name, input_shapes, data_reader, config)