Searched defs:test_where (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_masked.py | 323 def test_where(self, sparse_kind, fill_value): member in TestMasked |
| H A D | test_maskedtensor.py | 182 def test_where(self, device): member in TestBasics |
| H A D | test_tensorexpr.py | 1451 def test_where(self): member in TestTensorExprFuser |
| H A D | test_mps.py | 7812 def test_where(self): member in TestMPS |
| H A D | test_jit.py | 11248 def test_where(self): member in TestScript |
| /aosp_15_r20/external/executorch/kernels/test/ | ||
| H A D | op_where_test.cpp | 36 void test_where() { in test_where() function in OpWhereOutTest |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_vmap.py | 3848 def test_where(self, device): member in TestVmapBatchedGradient |
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_pytorch_onnx_onnxruntime.py | 9790 def test_where(self): member in TestONNXRuntime |