Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
caffe2
operators
softmax_shared.h
1
#ifndef CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
2
#define CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
3
4
#include "caffe2/core/context.h"
5
#include "caffe2/core/operator.h"
6
7
namespace
caffe2
{
8
9
void
SoftmaxCPU(
10
CPUContext& context,
11
const
int
N,
12
const
int
D,
13
const
float
* Xdata,
14
float
* Ydata,
15
float
* scale,
16
const
float
* sum_multiplier,
17
bool
logarithmic,
18
float
* rowmax);
19
}
// namespace caffe2
20
21
#endif // #define CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
convert_encoded_to_raw_leveldb.cc:47
Generated on Thu Apr 19 2018 13:03:56 for Caffe2 - C++ API by
1.8.11