| | |
|---|
| | | int batch; |
|---|
| | | int inputs; |
|---|
| | | float probability; |
|---|
| | | #ifdef GPU |
|---|
| | | float scale; |
|---|
| | | float *rand; |
|---|
| | | float *output; |
|---|
| | | #ifdef GPU |
|---|
| | | cl_mem rand_cl; |
|---|
| | | cl_mem output_cl; |
|---|
| | | #endif |
|---|
| | | } dropout_layer; |
|---|
| | | |
|---|
| | | dropout_layer *make_dropout_layer(int batch, int inputs, float probability); |
|---|
| | | |
|---|
| | | void forward_dropout_layer(dropout_layer layer, float *input); |
|---|
| | | void backward_dropout_layer(dropout_layer layer, float *input, float *delta); |
|---|
| | | #ifdef GPU |
|---|
| | | void backward_dropout_layer(dropout_layer layer, float *delta); |
|---|
| | | |
|---|
| | | #ifdef GPU |
|---|
| | | void forward_dropout_layer_gpu(dropout_layer layer, cl_mem input); |
|---|
| | | void backward_dropout_layer_gpu(dropout_layer layer, cl_mem delta); |
|---|
| | | |
|---|
| | | #endif |
|---|
| | | #endif |
|---|
