Fork me on GitHub

src/arraymancer/nn_primitives/backend/nnpack_interface

  Source Edit

Procs

proc nnpack_conv2d(input, weight, bias: Tensor[float32]; padding, stride: Size2D): Tensor[
    float32] {.noinit, ...raises: [ValueError], tags: [], forbids: [].}
  Source Edit
proc nnpack_conv2d_gradient[T](input, weight: Tensor[float32];
                               padding, stride: Size2D; grad_output: Tensor[T];
                               grad_input, grad_weight: var Tensor[T])
  Source Edit
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood