diff options
Diffstat (limited to 'include/nn.h')
| -rw-r--r-- | include/nn.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/include/nn.h b/include/nn.h index 3a2277a..1d3eba0 100644 --- a/include/nn.h +++ b/include/nn.h @@ -41,6 +41,8 @@ int nn_last_cell_idxs(uint8_t *out, tak_state_p state); // Network // =================================================================== +typedef float (*nn_function)(tak_state_p); + float eval_net(const float *board, const unsigned int n_inp, const float *w, const float *biases, const float *logit_w, const float logit_b, const unsigned int n_dense, |
