• 2-D convolution with separable filters.

    Performs a depthwise convolution that acts separately on channels followed by a pointwise convolution that mixes channels. Note that this is separability between dimensions [1, 2] and 3, not spatial separability between dimensions 1 and 2.

    See https://www.tensorflow.org/api_docs/python/tf/nn/separable_conv2d for more details.

    Type Parameters

    Parameters

    • x: TensorLike | T

      The input tensor, of rank 4 or rank 3, of shape [batch, height, width, inChannels]. If rank 3, batch of 1 is assumed.

    • depthwiseFilter: Tensor4D | TensorLike

      The depthwise filter tensor, rank 4, of shape [filterHeight, filterWidth, inChannels, channelMultiplier]. This is the filter used in the first step.

    • pointwiseFilter: Tensor4D | TensorLike

      The pointwise filter tensor, rank 4, of shape [1, 1, inChannels * channelMultiplier, outChannels]. This is the filter used in the second step.

    • strides: number | [number, number]

      The strides of the convolution: [strideHeight, strideWidth]. If strides is a single number, then strideHeight == strideWidth.

    • pad: "valid" | "same"

      The type of padding algorithm.

    • Optionaldilation: number | [number, number]
    • OptionaldataFormat: "NHWC" | "NCHW"

    Returns T