31 lines
864 B
Text
31 lines
864 B
Text
/*
|
|
* File: cpuFuncs.h
|
|
* Author: Alex Krizhevsky
|
|
*
|
|
* Created on September 10, 2012, 5:05 PM
|
|
*/
|
|
|
|
#ifndef CPUFUNCS_H
|
|
#define CPUFUNCS_H
|
|
#include <helper_cuda.h>
|
|
#include <softmaxtree.cuh>
|
|
/*
|
|
* weights: (numNodes, numFeatures)
|
|
* nodes: numNodesAtDepth-length array of ushort2
|
|
* where x coordinate gives node idx and y coordinate gives parent idx
|
|
* targets: (numNodes, numFeatures)
|
|
*
|
|
*/
|
|
void cpuSoftmaxTreeFwd(float* weights, float* targets, const int numFeatures, SoftmaxTree& tree);
|
|
|
|
/*
|
|
* grads: (numNodes, numFeatures)
|
|
*
|
|
*/
|
|
void cpuSoftmaxTreeBwd(float* grads, const int numFeatures, SoftmaxTree& tree);
|
|
|
|
void cpuSoftmaxTreeUpdateWeights(float* weights, float* weightsInc, float* weightsGrad,
|
|
const int numFeatures, float eps, const float mom, float wc, SoftmaxTree& tree);
|
|
|
|
#endif /* CPUFUNCS_H */
|
|
|