xlnstorch.csrc.conv3d_forward

Contents

xlnstorch.csrc.conv3d_forward#

xlnstorch.csrc.conv3d_forward(x, weight, bias, base, stride_d, stride_h, stride_w, padding_d, padding_h, padding_w, dilation_d, dilation_h, dilation_w, groups)#

Perform LNS 3D convolution.

Parameters:

  • x (torch.Tensor) – The input tensor in LNS format (torch.int64).

  • weight (torch.Tensor) – The convolution kernel in LNS format (torch.int64).

  • bias (torch.Tensor) – The bias tensor in LNS format (torch.int64).

  • base (torch.Tensor) – The base for the LNS representation.

  • stride_d (int) – The depth stride of the convolution.

  • stride_h (int) – The height stride of the convolution.

  • stride_w (int) – The width stride of the convolution.

  • padding_d (int) – The depth padding applied to the input tensor.

  • padding_h (int) – The height padding applied to the input tensor.

  • padding_w (int) – The width padding applied to the input tensor.

  • dilation_d (int) – The depth dilation factor for the convolution.

  • dilation_h (int) – The height dilation factor for the convolution.

  • dilation_w (int) – The width dilation factor for the convolution.

  • groups (int) – The number of groups for the convolution.

Returns:

The output tensor in LNS format (torch.int64) after convolution.

Return type:

torch.Tensor

Contents