Google Git
Sign in
android / platform / external / pytorch / b90a9c7db2bfcbdaf6910f6eefea8618a5cb53cb / . / benchmarks / cpp / nvfuser
tree: b265dc22d8f394b21613f24607e370051c8a0d27 [path history] [tgz]
  1. batch_norm_channels_first.cpp
  2. batch_norm_channels_first_backward.cpp
  3. batch_norm_channels_last.cpp
  4. batch_norm_channels_last_backward.cpp
  5. bert.cpp
  6. broadcast.cpp
  7. CMakeLists.txt
  8. gelu_backward.cpp
  9. heuristic_cache.cpp
  10. heuristic_lookup.cpp
  11. instance_norm.cpp
  12. layer_norm.cpp
  13. layer_norm_backward.cpp
  14. lstm_cell.cpp
  15. main.cpp
  16. matmul.cpp
  17. reduction.cpp
  18. rms_norm.cpp
  19. rms_norm_backward.cpp
  20. scale_bias_relu.cpp
  21. shape_inference.cpp
  22. softmax.cpp
  23. softmax_backward.cpp
  24. softmax_dropout.cpp
  25. timm.cpp
  26. transpose.cpp
  27. utils.cpp
  28. utils.h