Searched refs:test_masked_fill_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 3388 void test_masked_fill_tensor( in test_masked_fill_tensor() function
3411 test_masked_fill_tensor({3, 4, 2, 3}, {1, 4, 1, 1}); in TEST_F()
3415 test_masked_fill_tensor({3, 5, 2, 3}, {1, 5, 1, 1}); in TEST_F()
|