Home
last modified time | relevance | path

Searched refs:a_set_with_number (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtensor_indexing.cpp594 auto a_set_with_number = a.clone(); in TEST() local
598 a_set_with_number.index_put_({0}, b); in TEST()
600 assert_tensor_equal(a_set_with_number, a_set_with_scalar); in TEST()
/aosp_15_r20/external/pytorch/test/
H A Dtest_indexing.py1319 a_set_with_number = a.clone()
1323 a_set_with_number[0] = b
1325 self.assertEqual(a_set_with_number, a_set_with_scalar)
H A Dtest_mps.py11584 a_set_with_number = a.clone()
11588 a_set_with_number[0] = b
11590 self.assertEqual(a_set_with_number, a_set_with_scalar)