pytorch/benchmarks/static_runtime
Mike Iovine 2ca66ffb7d [SR] Force split_and_squeeze usage via graph transformation (#74274)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74274

Reviewed By: navahgar

Differential Revision: D34913889

fbshipit-source-id: 655d3f1e5f4c027cb94758b74826a4b4882e9458
(cherry picked from commit bc94d30b69888ca6633a27090a3b87a08919231a)
2022-03-29 19:13:40 +00:00
..
CMakeLists.txt [Static Runtime] Add auto-generated out variant dispatchers (#72603) 2022-02-28 08:39:10 +00:00
deep_wide_pt.cc
deep_wide_pt.h
deep_wide_pt_bench.cc
test_cpu_fusion.cc [Static Runtime] Add test that runs NNC fused kernels in parallel (#73256) 2022-02-28 17:44:45 +00:00
test_generated_ops.cc [Static Runtime] Fix a broken test & Add an out variant wrapper for mse_loss (#73574) 2022-03-02 04:36:31 +00:00
test_static_module.cc [SR] Force split_and_squeeze usage via graph transformation (#74274) 2022-03-29 19:13:40 +00:00
test_static_runtime.cc [SR] Native implementation for reshape_as (#74585) 2022-03-28 17:02:14 +00:00
test_utils.cc [Static Runtime] Enable all tests to run with TensorExpr fuser (#73263) 2022-02-24 00:34:34 +00:00
test_utils.h [SR] Graph pass to create owned refs of special IValues (#69835) 2022-02-02 19:30:50 +00:00