| | |
|---|
| | | int batch; |
|---|
| | | float *delta; |
|---|
| | | float *output; |
|---|
| | | float *jacobian; |
|---|
| | | #ifdef GPU |
|---|
| | | float * delta_gpu; |
|---|
| | | float * output_gpu; |
|---|
| | | #endif |
|---|
| | | } softmax_layer; |
|---|
| | | |
|---|
| | | softmax_layer *make_softmax_layer(int batch, int inputs); |
|---|
| | | void forward_softmax_layer(const softmax_layer layer, float *input); |
|---|
| | | void backward_softmax_layer(const softmax_layer layer, float *input, float *delta); |
|---|
| | | void backward_softmax_layer(const softmax_layer layer, float *delta); |
|---|
| | | |
|---|
| | | #ifdef GPU |
|---|
| | | void pull_softmax_layer_output(const softmax_layer layer); |
|---|
| | | void forward_softmax_layer_gpu(const softmax_layer layer, float *input); |
|---|
| | | void backward_softmax_layer_gpu(const softmax_layer layer, float *delta); |
|---|
| | | #endif |
|---|
| | | |
|---|
| | | #endif |
|---|
