aboutsummaryrefslogtreecommitdiff
path: root/include/nn.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/nn.h')
-rw-r--r--include/nn.h2
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,