def conv2d_bias_layout

stelliferous@0.0.2.0 · tensor_convolution.bend:342

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

342def conv2d_bias_layout(~custom: F32 -> F32,+activation: Activations.Activation,dense: Bool,input: Tensors.Tensor<F32>,weights: Tensors.Tensor<F32>,bias: Tensors.Tensor<F32>,stride: U32,padding: U32) -> Tensors.Tensor<F32>:
343  match bias:
344    case Tensors.Tensor{storage,shape,fill,status,workers}:
345      conv2d_selected(~custom,activation,dense,input,weights,Tensors.Tensor{storage,shape,fill,status,workers},stride,padding)
346    case Tensors.Banded{bands,geometry,shape,fill,status,workers}:
347      conv2d_selected(~custom,activation,True{},input,weights,Tensors.Banded{bands,geometry,shape,fill,status,workers},stride,padding)