Google Git
Sign in
android / platform / external / pytorch / d1be36ceab / . / benchmarks / cpp / nvfuser
tree: 7f7b265d89d7c697bdb5d8408024efdfe8803121 [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. reduction.cpp
  17. rms_norm.cpp
  18. rms_norm_backward.cpp
  19. scale_bias_relu.cpp
  20. shape_inference.cpp
  21. softmax.cpp
  22. softmax_backward.cpp
  23. softmax_dropout.cpp
  24. timm.cpp
  25. transpose.cpp
  26. utils.cpp
  27. utils.h