PoolRefImpl

reference implementaion of pooling function

struct PoolRefImpl (

alias poolFun

) {

import mir.ndslice;;

import numir;;

static auto forward(Variable!(T, tensorDims, HostStorage) x, int[poolDims] windowA, int[poolDims] padA, int[poolDims] strideA);

static void backward(Variable!(T, tensorDims, HostStorage) gx, Variable!(T, tensorDims, HostStorage) x, Variable!(T, tensorDims, HostStorage) gy, Variable!(T, tensorDims, HostStorage) y, int[poolDims] windowA, int[poolDims] padA, int[poolDims] strideA);

}

Members

Static functions

backward void backward(Variable!(T, tensorDims, HostStorage) gx, Variable!(T, tensorDims, HostStorage) x, Variable!(T, tensorDims, HostStorage) gy, Variable!(T, tensorDims, HostStorage) y, int[poolDims] windowA, int[poolDims] padA, int[poolDims] strideA): Undocumented in source. Be warned that the author may not have intended to support it.
forward auto forward(Variable!(T, tensorDims, HostStorage) x, int[poolDims] windowA, int[poolDims] padA, int[poolDims] strideA): Undocumented in source. Be warned that the author may not have intended to support it.

Meta

Source

See Implementation

grain functions unary aliases

functions

manifest constants

CUDNN_ACTIVATION_IMPL_MIXIN

structs