Searched defs:at_tensor (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | tensor_interop_test.cpp | 15 at::Tensor at_tensor(c2_tensor); in TEST() local
29 at::Tensor at_tensor(c2_tensor); in TEST() local
48 at::Tensor at_tensor(c2_tensor); in TEST() local
67 at::Tensor at_tensor(c2_tensor); in TEST() local
115 at::Tensor at_tensor(c2_tensor); in TEST() local
228 auto at_tensor = at::ones({5, 5}, at::dtype(at::kFloat)); in TEST() local
258 auto at_tensor = at::ones({5, 5}, at::dtype(at::kFloat)).t(); in TEST() local
269 auto at_tensor = at::zeros({2, 5}, at::dtype(at::kFloat)); in TEST() local
287 at::Tensor at_tensor = in TEST() local
305 at::Tensor at_tensor(c2_tensor); in TEST() local
[all …]
|
| H A D | cuda_tensor_interop_test.cpp | 33 at::Tensor at_tensor(c2_tensor); in TEST() local
52 at::Tensor at_tensor(c2_tensor); in TEST() local
71 at::Tensor at_tensor(c2_tensor); in TEST() local
138 auto at_tensor = at::ones({5, 5}, at::dtype(at::kFloat).device(at::kCUDA)); in TEST() local
|
| /aosp_15_r20/external/executorch/extension/aten_util/test/ |
| H A D | aten_bridge_test.cpp | 35 auto at_tensor = generate_at_tensor(); in TEST() local
55 auto at_tensor = generate_at_tensor(); in TEST() local
83 auto at_tensor = generate_at_tensor(); in TEST() local
108 auto at_tensor = generate_at_tensor(); in TEST() local
131 auto at_tensor = generate_at_tensor(); in TEST() local
152 auto at_tensor = generate_at_tensor(); in TEST() local
|
| /aosp_15_r20/external/executorch/extension/training/pybindings/ |
| H A D | _training_lib.cpp | 88 auto at_tensor = pair.second; in step() local
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/runtime/ |
| H A D | test_runtime_utils.cpp | 57 auto at_tensor = at::full({3, 5}, 4.f); in TEST() local
|
| /aosp_15_r20/external/executorch/extension/pybindings/ |
| H A D | pybindings.cpp | 697 auto at_tensor = python_input.cast<at::Tensor>(); in run_method() local
|