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.

indexing_one_hot.h 305 B

1234567891011121314
  1. #pragma once
  2. #include "megdnn/handle.h"
  3. namespace megdnn {
  4. namespace test {
  5. void run_indexing_one_hot_test(
  6. Handle* handle, const thin_function<void()>& fail_test = {});
  7. void run_indexing_set_one_hot_test(Handle* handle);
  8. } // namespace test
  9. } // namespace megdnn
  10. // vim: syntax=cpp.doxygen