mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
22 lines
462 B
C++
22 lines
462 B
C++
#ifndef CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
|
|
#define CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
|
|
|
|
#include "caffe2/core/context.h"
|
|
#include "caffe2/core/operator.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
void SoftmaxCPU(
|
|
CPUContext& context,
|
|
const int N,
|
|
const int D,
|
|
const float* Xdata,
|
|
float* Ydata,
|
|
float* scale,
|
|
const float* sum_multiplier,
|
|
bool logarithmic,
|
|
float* rowmax);
|
|
} // namespace caffe2
|
|
|
|
#endif // #define CAFFE2_OPERATORS_SOFTMAX_SHARED_H_
|