.. _f-opconvhelper: module ``onnxrt.ops_cpu.op_conv_helper`` ======================================== Short summary +++++++++++++ module ``mlprodict.onnxrt.ops_cpu.op_conv_helper`` Helpers for operators Conv, ConvTranspose. :githublink:`%|py|7` Functions +++++++++ +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | function | truncated documentation | +===========================================================================================================+==========================================================================================================================+ | :func:`_get_indices ` | | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`_is_out ` | | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`col2im_indices ` | Source `im2col.py `_. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`col2im_nchw ` | C implementation of a partial col2im. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`get_im2col_indices ` | Source `im2col.py `_. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`im2col ` | Returns the result of `im2col` on a image `NHCW` where N is 1. The function is equivalent to :func:`torch.nn.Unfold` ... | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`im2col_indices ` | Source `im2col.py `_. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`im2col_infer_output_shape ` | Computes the ouput shape of im2col. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`im2col_naive_implementation ` | Naive implementation for `im2col` or :func:`torch.nn.Unfold` (but with `padding=1`). | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`im2col_nchw ` | C implementation of a partial im2col. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`im2col_nn ` | Functions :func:`nn_im2col_2d` and :func:`im2col` returns the same results but with different shapes. This function ... | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`im2col_recursive ` | Recursive implementation, falls back to :func:`im2col_naive_implementation` for dimension `<= fall_back_dim`. The ... | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`new_array ` | Creates a new empty array. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`nn_col2im_2d ` | C++ implementation for `col2im` or :func:`torch.nn.Fold`. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ | :func:`nn_im2col_2d ` | C++ implementation for `im2col` or :func:`torch.nn.Unfold`. | +-----------------------------------------------------------------------------------------------------------+--------------------------------------------------------------------------------------------------------------------------+ Documentation +++++++++++++ .. automodule:: mlprodict.onnxrt.ops_cpu.op_conv_helper :members: :special-members: __init__ :show-inheritance: