You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

get_params.template 1.0 kB

12345678910111213141516171819202122232425262728293031
  1. #include "src/cuda/convolution/cudnn_heuristic.h"
  2. #include "megdnn.h"
  3. using namespace megdnn;
  4. using namespace cuda;
  5. using namespace convolution;
  6. bool convolution::heuristic_params_available(
  7. int cuda_major, int cuda_minor, size_t* layer_num_p,
  8. const size_t** layers_dim_p, const float** matrices_p,
  9. const float** biases_p, const float** alpha_p, const float** beta_p,
  10. const ConvolutionType& conv_type, float** hidden_units_p,
  11. float** time_pred_p, float** mask_p) {
  12. MEGDNN_MARK_USED_VAR(cuda_major);
  13. MEGDNN_MARK_USED_VAR(cuda_minor);
  14. MEGDNN_MARK_USED_VAR(layer_num_p);
  15. MEGDNN_MARK_USED_VAR(layers_dim_p);
  16. MEGDNN_MARK_USED_VAR(matrices_p);
  17. MEGDNN_MARK_USED_VAR(biases_p);
  18. MEGDNN_MARK_USED_VAR(alpha_p);
  19. MEGDNN_MARK_USED_VAR(beta_p);
  20. MEGDNN_MARK_USED_VAR(conv_type);
  21. MEGDNN_MARK_USED_VAR(hidden_units_p);
  22. MEGDNN_MARK_USED_VAR(time_pred_p);
  23. MEGDNN_MARK_USED_VAR(mask_p);
  24. {cudnn_select}
  25. return false;
  26. }
  27. // vim: syntax=cpp.doxygen