• Jesse Grabowski's avatar
    Implement `Convolve2D` Op (#1397) · 10d225dd
    Jesse Grabowski 提交于
    * Implement simple wrapper Op for `scipy.signal.convolve2d`
    
    * Better shape inference
    
    * conv2d gradient v0
    
    * Implement Convolve2d and gradients
    
    * Add same mode, and FFT support
    
    * Relax test tolerance in float32
    
    * Change seed
    
    * Relax test tolerance in float32
    
    * Add `convolve2d` to `signal.__init__`
    
    ---------
    Co-authored-by: 's avatarricardoV94 <ricardo.vieira1994@gmail.com>
    10d225dd
test_conv.py 7.7 KB