BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
Classes | Namespaces | Functions
convolution.h File Reference
#include "layer_base.h"
Include dependency graph for convolution.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  bc::nn::Convolution< SystemTag, ValueType, Optimizer >
 

Namespaces

 bc
 The Evaluator determines if an expression needs to be greedily optimized.
 
 bc::nn
 

Functions

template<class SystemTag = BLACKCAT_DEFAULT_SYSTEM_T, class Optimizer = nn_default_optimizer_type>
auto bc::nn::convolution (SystemTag system_tag, Dim< 3 > img_dims, Dim< 3 > krnl_dims, Dim< 2 > padding=Dim< 2 >().fill(0), Dim< 2 > strides=Dim< 2 >().fill(1), Dim< 2 > dilation=Dim< 2 >().fill(1), Optimizer=Optimizer())
 
template<class SystemTag = BLACKCAT_DEFAULT_SYSTEM_T, class Optimizer = nn_default_optimizer_type>
auto bc::nn::recurrent_convolution (SystemTag system_tag, Dim< 3 > img_dims, Dim< 3 > krnl_dims, Dim< 2 > padding=Dim< 2 >().fill(0), Dim< 2 > strides=Dim< 2 >().fill(1), Dim< 2 > dilation=Dim< 2 >().fill(1), Optimizer=Optimizer())
 
template<class SystemTag , class Optimizer >
auto bc::nn::convolution (SystemTag system_tag, Dim< 3 > img_dims, Dim< 3 > krnl_dims, Optimizer, Dim< 2 > padding=Dim< 2 >().fill(0), Dim< 2 > strides=Dim< 2 >().fill(1), Dim< 2 > dilation=Dim< 2 >().fill(1))
 
template<class SystemTag , class Optimizer >
auto bc::nn::recurrent_convolution (SystemTag system_tag, Dim< 3 > img_dims, Dim< 3 > krnl_dims, Optimizer, Dim< 2 > padding=Dim< 2 >().fill(0), Dim< 2 > strides=Dim< 2 >().fill(1), Dim< 2 > dilation=Dim< 2 >().fill(1))