dcnn.h

   1
   2 #ifndef PACHI_DCNN_H
   3 #define PACHI_DCNN_H
   4
   5 #ifdef __cplusplus
   6 extern "C" {
   7 #endif
   8
   9
  10 #ifdef DCNN
  11
  12 #define DCNN_BEST_N 5
  13
  14 void dcnn_get_moves(struct board *b, enum stone color, float result[]);
  15 bool using_dcnn(struct board *b);
  16 void dcnn_quiet_caffe(int argc, char *argv[]);
  17 void dcnn_init();
  18 void find_dcnn_best_moves(struct board *b, float *r, coord_t *best, float *best_r);
  19 void print_dcnn_best_moves(struct tree_node *node, struct board *b, coord_t *best, float *best_r);
  20 struct engine *engine_dcnn_init(char *arg, struct board *b);
  21
  22 #else
  23
  24 #define using_dcnn(b)  0
  25 #define dcnn_quiet_caffe(argc, argv)
  26 #define dcnn_init()
  27
  28 #endif
  29
  30
  31 #ifdef __cplusplus
  32 }
  33 #endif
  34
  35 #endif /* PACHI_DCNN_H */