Caffe2 - C++ API
A deep learning, cross platform ML framework
softmax_shared.h
1 
17 #ifndef CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
18 #define CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
19 
20 #include "caffe2/core/context.h"
21 #include "caffe2/core/operator.h"
22 
23 namespace caffe2 {
24 
25 void SoftmaxCPU(
26  CPUContext& context,
27  const int N,
28  const int D,
29  const float* Xdata,
30  float* Ydata,
31  float* scale,
32  const float* sum_multiplier,
33  bool logarithmic,
34  float* rowmax);
35 } // namespace caffe2
36 
37 #endif // #define CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
Copyright (c) 2016-present, Facebook, Inc.