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_einsum_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.core.parse_layernorm_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- 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.pquant module
- hls4ml.converters.pytorch.pquant.add_quantizer_info(class_object, input_names, input_shapes, output_shape, layer)
- hls4ml.converters.pytorch.pquant.make_pquant_handler(base_parse_func, op, op_check=None)
- hls4ml.converters.pytorch.pquant.parse_pqactivation_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_pqbatchnorm1d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_pqbatchnorm2d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_pqconv1d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_pqconv2d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_pqlinear_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_pqpool1d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_pqpool2d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_quant_activation_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.pquant.parse_quantizer_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_constantpad1d_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.reshape.parse_constantpad2d_layer(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)
hls4ml.converters.pytorch.snn module
- hls4ml.converters.pytorch.snn.parse_lif_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)
- hls4ml.converters.pytorch.snn.parse_snn_readout_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config)