Searched defs:mask_cpu (Results 1 – 1 of 1) sorted by relevance
3212 const auto mask_cpu = in TEST_F() local3229 const auto mask_cpu = at::randint( in TEST_F() local3246 const auto mask_cpu = in TEST_F() local3263 const auto mask_cpu = in TEST_F() local3355 at::Tensor mask_cpu = at::randint( in test_masked_fill_scalar() local3395 at::Tensor mask_cpu = in test_masked_fill_tensor() local