Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtensor_indexing.cpp595 auto a_set_with_scalar = a.clone(); in TEST() local
599 a_set_with_scalar.index_put_({zero}, 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.py1320 a_set_with_scalar = a.clone()
1324 a_set_with_scalar[zero] = b
1325 self.assertEqual(a_set_with_number, a_set_with_scalar)
H A Dtest_mps.py11585 a_set_with_scalar = a.clone()
11589 a_set_with_scalar[zero] = b
11590 self.assertEqual(a_set_with_number, a_set_with_scalar)