Home
last modified time | relevance | path

Searched refs:test_dynamic_shapes (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A DCMakeLists.txt9 ${TENSOREXPR_TEST_ROOT}/test_dynamic_shapes.cpp
/aosp_15_r20/external/pytorch/.ci/docker/
H A Drequirements-ci.txt145 #test that import: test_vmap.py, test_aotdispatch.py, test_dynamic_shapes.py,
/aosp_15_r20/external/pytorch/.ci/pytorch/
H A Dtest.sh1387 … --include test_modules test_mkldnn test_mkldnn_fusion test_openmp test_torch test_dynamic_shapes \
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cuda_repro.py253 def test_dynamic_shapes(self): member in CudaReproTests
H A Dtest_compiled_autograd.py745 def test_dynamic_shapes(self): member in TestCompiledAutograd
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit_fuser_te.py2463 def test_dynamic_shapes(self): member in TestTEFuser