def conv2d

stelliferous@0.0.2.0 · tensor_convolution.bend:349

import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/math_activation.bend as Activations

349def conv2d(~custom: F32 -> F32,+activation: Activations.Activation,input: Tensors.Tensor<F32>,weights: Tensors.Tensor<F32>,bias: Tensors.Tensor<F32>,stride: U32,padding: U32) -> Tensors.Tensor<F32>:
350  match weights:
351    case Tensors.Tensor{storage,shape,fill,status,workers}:
352      conv2d_bias_layout(~custom,activation,False{},input,Tensors.Tensor{storage,shape,fill,status,workers},bias,stride,padding)
353    case Tensors.Banded{bands,geometry,shape,fill,status,workers}:
354      conv2d_bias_layout(~custom,activation,True{},input,Tensors.Banded{bands,geometry,shape,fill,status,workers},bias,stride,padding)