hls4ml.converters.pytorch package
Submodules
hls4ml.converters.pytorch.convolution module
- hls4ml.converters.pytorch.convolution.parse_conv1d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.convolution.parse_conv2d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
hls4ml.converters.pytorch.core module
- hls4ml.converters.pytorch.core.parse_activation_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.core.parse_batchnorm_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.core.parse_constant_layer(operation, layer_name, node)
- hls4ml.converters.pytorch.core.parse_linear_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
hls4ml.converters.pytorch.merge module
- hls4ml.converters.pytorch.merge.parse_concat_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.merge.parse_merge_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
hls4ml.converters.pytorch.pooling module
- hls4ml.converters.pytorch.pooling.parse_pooling_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
hls4ml.converters.pytorch.recurrent module
- hls4ml.converters.pytorch.recurrent.parse_rnn_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
hls4ml.converters.pytorch.reshape module
- hls4ml.converters.pytorch.reshape.handle_upsample(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.reshape.parse_flatten_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.reshape.parse_reshape_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.reshape.parse_squeeze_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.reshape.parse_unsqueeze_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)