def conv2d

stelliferous@0.0.2.0 · tensor_f32.bend:277

import Base
import stelliferous@0.0.2.0/math_activation.bend as Activations
import stelliferous@0.0.2.0/tensor_convolution.bend as TensorConvolution

277# Ordered convolution over CHW or NCHW; the activation runs on each finished
278# leaf while it is in cache.
279def conv2d(input: Tensor(),weights: Tensor(),bias: Tensor(),stride: U32,padding: U32,activation: Activations.Activation) -> Tensor():
280  TensorConvolution.conv2d(~Activations.identity_value,activation,input,weights,bias,stride,padding)