
    6bi$                         S r SSKJs  Jr  SSKJr  SSKJr  SSKJr  SSKJ	r	  SSK
Jr  SSKJr  SS	KJr  \" S
S5       " S S\5      5       r\rg)z)Keras depthwise separable 1D convolution.    N)activations)constraints)initializers)regularizers)SeparableConv)
conv_utils)keras_exportzkeras.layers.SeparableConv1Dz#keras.layers.SeparableConvolution1Dc                   X   ^  \ rS rSrSr                 SU 4S jjrS rSrU =r$ )SeparableConv1D   a  Depthwise separable 1D convolution.

This layer performs a depthwise convolution that acts separately on
channels, followed by a pointwise convolution that mixes channels.
If `use_bias` is True and a bias initializer is provided,
it adds a bias vector to the output.
It then optionally applies an activation function to produce the final
output.

Args:
  filters: Integer, the dimensionality of the output space (i.e. the number
    of filters in the convolution).
  kernel_size: A single integer specifying the spatial
    dimensions of the filters.
  strides: A single integer specifying the strides
    of the convolution.
    Specifying any `stride` value != 1 is incompatible with specifying
    any `dilation_rate` value != 1.
  padding: One of `"valid"`, `"same"`, or `"causal"` (case-insensitive).
    `"valid"` means no padding. `"same"` results in padding with zeros
    evenly to the left/right or up/down of the input such that output has
    the same height/width dimension as the input. `"causal"` results in
    causal (dilated) convolutions, e.g. `output[t]` does not depend on
    `input[t+1:]`.
  data_format: A string, one of `channels_last` (default) or
    `channels_first`.  The ordering of the dimensions in the inputs.
    `channels_last` corresponds to inputs with shape
    `(batch_size, length, channels)` while `channels_first` corresponds to
    inputs with shape `(batch_size, channels, length)`.
  dilation_rate: A single integer, specifying
    the dilation rate to use for dilated convolution.
  depth_multiplier: The number of depthwise convolution output channels for
    each input channel. The total number of depthwise convolution output
    channels will be equal to `num_filters_in * depth_multiplier`.
  activation: Activation function to use.
    If you don't specify anything, no activation is applied
    (see `keras.activations`).
  use_bias: Boolean, whether the layer uses a bias.
  depthwise_initializer: An initializer for the depthwise convolution kernel
    (see `keras.initializers`). If None, then the default initializer
    ('glorot_uniform') will be used.
  pointwise_initializer: An initializer for the pointwise convolution kernel
    (see `keras.initializers`). If None, then the default initializer
    ('glorot_uniform') will be used.
  bias_initializer: An initializer for the bias vector. If None, the default
    initializer ('zeros') will be used (see `keras.initializers`).
  depthwise_regularizer: Optional regularizer for the depthwise
    convolution kernel (see `keras.regularizers`).
  pointwise_regularizer: Optional regularizer for the pointwise
    convolution kernel (see `keras.regularizers`).
  bias_regularizer: Optional regularizer for the bias vector
    (see `keras.regularizers`).
  activity_regularizer: Optional regularizer function for the output
    (see `keras.regularizers`).
  depthwise_constraint: Optional projection function to be applied to the
    depthwise kernel after being updated by an `Optimizer` (e.g. used for
    norm constraints or value constraints for layer weights). The function
    must take as input the unprojected variable and must return the
    projected variable (which must have the same shape). Constraints are
    not safe to use when doing asynchronous distributed training
    (see `keras.constraints`).
  pointwise_constraint: Optional projection function to be applied to the
    pointwise kernel after being updated by an `Optimizer`
    (see `keras.constraints`).
  bias_constraint: Optional projection function to be applied to the
    bias after being updated by an `Optimizer`
    (see `keras.constraints`).
  trainable: Boolean, if `True` the weights of this layer will be marked as
    trainable (and listed in `layer.trainable_weights`).

Input shape:
  3D tensor with shape:
  `(batch_size, channels, steps)` if data_format='channels_first'
  or 3D tensor with shape:
  `(batch_size, steps, channels)` if data_format='channels_last'.

Output shape:
  3D tensor with shape:
  `(batch_size, filters, new_steps)` if data_format='channels_first'
  or 3D tensor with shape:
  `(batch_size,  new_steps, filters)` if data_format='channels_last'.
  `new_steps` value might have changed due to padding or strides.

Returns:
  A tensor of rank 3 representing
  `activation(separableconv1d(inputs, kernel) + bias)`.
c                 V  > [         TU ]  " S0 SS_SU_SU_SU_SU_SU_SU_S	U_S
[        R                  " U5      _SU	_S[        R                  " U
5      _S[        R                  " U5      _S[        R                  " U5      _S[
        R                  " U5      _S[
        R                  " U5      _S[
        R                  " U5      _S[
        R                  " U5      _S[        R                  " U5      _S[        R                  " U5      _S[        R                  " U5      _UD6  g )Nrank   filterskernel_sizestridespaddingdata_formatdilation_ratedepth_multiplier
activationuse_biasdepthwise_initializerpointwise_initializerbias_initializerdepthwise_regularizerpointwise_regularizerbias_regularizeractivity_regularizerdepthwise_constraintpointwise_constraintbias_constraint )super__init__r   getr   r   r   )selfr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   kwargs	__class__s                        l/home/james-whalen/.local/lib/python3.13/site-packages/tf_keras/src/layers/convolutional/separable_conv1d.pyr%   SeparableConv1D.__init__{   sS   . 	 	
	
	
 $	
 		

 	
 $	
 (	
 .	
 #z2	
 	
 #/"2"23H"I	
 #/"2"23H"I	
 *--.>?	
 #/"2"23H"I	
 #/"2"23H"I	
  *--.>?!	
" ".!1!12F!G#	
$ "-1E!F%	
& "-1E!F'	
( (OOO<+	
    c                    U R                   S:X  a%  [        R                  " XR                  U5      5      nU R                  S:X  a  SU R
                  S-  -   S-   nSnOSU R
                  S-  -   nSn[        R                  " X5      n[        R                  " U R                  S5      n[        R                  " U R                  S5      nSU R                  -   nU R                   S:X  a  SnOU R                   n[        R                  R                  R                  R                  UUUUUR                  5       U[        R                   " U R                  S	S
9S9nU R"                  (       aF  [        R                  R%                  UU R&                  [        R                   " U R                  S	S
9S9n[        R(                  " X/5      nU R*                  b  U R+                  U5      $ U$ )Ncausalchannels_last)r      r   )r   r   r   valid   )ndim)r   r   rater   )r   )r   tfpad_compute_causal_paddingr   r   expand_dimsdepthwise_kernelpointwise_kernelr   compatv1nnseparable_conv2dupperr   convert_data_formatr   bias_addbiassqueezer   )	r'   inputsr   spatial_start_dimr9   r:   r   
op_paddingoutputss	            r*   callSeparableConv1D.call   s   <<8#VVF$@$@$HIF.T\\A--4G !t||a//G !
 :>>$*?*?C>>$*?*?Ct111<<8# JJ)),,//22$$&"66  q 3 

 ==eenn		&::$$1 % G **W&9:??&??7++r,   r#   )r   r1   Nr   r   NTglorot_uniformrJ   zerosNNNNNNN)	__name__
__module____qualname____firstlineno____doc__r%   rH   __static_attributes____classcell__)r)   s   @r*   r   r      sO    Vx .. ""!!!)-
^/ /r,   r   )rP   tensorflow.compat.v2r;   v2r5   tf_keras.srcr   r   r   r   5tf_keras.src.layers.convolutional.base_separable_convr   tf_keras.src.utilsr    tensorflow.python.util.tf_exportr	   r   SeparableConvolution1Dr#   r,   r*   <module>rZ      s[    0 " ! $ $ % % O ) : "$Iwm wwx ) r,   