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 9e576c1..871bd0f 100644 --- a/include/nn.h +++ b/include/nn.h @@ -35,6 +35,8 @@ void nn_prepare_head(float out_head[3], tak_state_p state); void nn_prepare_input(float *cur_board, tak_state_p state, const float embedding[NN_N_EMB]); + +int nn_last_cell_idxs(uint8_t *out, tak_state_p state); // =================================================================== // Network // =================================================================== |