def uses_band_windows

stelliferous@0.0.2.0 · tensor_convolution.bend:174

import stelliferous@0.0.2.0/tensor_convolution.bend as TensorConvolution
import Base
import stelliferous@0.0.2.0/kernel_shape.bend as KernelShape

174# Padding-only output windows and an empty batch use the existing dense path.
175# This is a representation choice, not a restriction on valid convolution shapes.
176def uses_band_windows(geometry: Geometry) -> Bool:
177  Geometry{count,batched,KernelShape.Dimensions{ci,rows,h,w,k,s,p}} = geometry
178  U32.is_lt(0,count) && Nat.is_lt(p,k)

Uses (6)

Used by (2)