FasTensor  1.0.0
Transform Supercomputing for AI
Macros | Functions | Variables
conv.cpp File Reference
#include <iostream>
#include <stdarg.h>
#include <vector>
#include <stdlib.h>
#include "ft.h"

Macros

#define KERNELS   3
 
#define KERNEL_HEIGHT   2
 
#define KERNEL_WIDTH   2
 

Functions

Stencil< vector< float > > UDF_Cov (const Stencil< float > &iStencil)
 
Stencil< vector< float > > UDF_Cov2 (const Stencil< float > &iStencil)
 
int main (int argc, char *argv[])
 

Variables

float kernel_weight [KERNELS][KERNEL_HEIGHT][KERNEL_WIDTH]
 

Macro Definition Documentation

◆ KERNEL_HEIGHT

#define KERNEL_HEIGHT   2

◆ KERNEL_WIDTH

#define KERNEL_WIDTH   2

◆ KERNELS

#define KERNELS   3

Function Documentation

◆ main()

int main ( int  argc,
char *  argv[] 
)

◆ UDF_Cov()

Stencil<vector<float> > UDF_Cov ( const Stencil< float > &  iStencil)

◆ UDF_Cov2()

Stencil<vector<float> > UDF_Cov2 ( const Stencil< float > &  iStencil)

Variable Documentation

◆ kernel_weight

float kernel_weight[KERNELS][KERNEL_HEIGHT][KERNEL_WIDTH]
Initial value:
= {
{{1, 2}, {3, 4}},
{{2, 3}, {4, 5}},
{{3, 4}, {5, 6}},
}