Searched defs:at_tensor_a (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | tensor_interop_test.cpp | 146 auto at_tensor_a = at::ones({5, 5}, at::dtype(at::kFloat)); in TEST() local
186 auto at_tensor_a = at::ones({5, 5}, at::dtype(at::kFloat)); in TEST() local
214 auto at_tensor_a = at::ones({5, 5}, at::dtype(at::kFloat)); in TEST() local
|
| H A D | cuda_tensor_interop_test.cpp | 82 auto at_tensor_a = at::ones({5, 5}, at::dtype(at::kFloat).device(at::kCUDA)); in TEST() local
123 auto at_tensor_a = at::ones({5, 5}, at::dtype(at::kFloat).device(at::kCUDA)); in TEST() local
|