Searched refs:managed_tensor_name_to_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_module.cc | 1340 FastMap<std::string, at::Tensor> managed_tensor_name_to_tensor, in testAssignStorageToManagedTensors() argument
1349 for (auto& key_value : managed_tensor_name_to_tensor) { in testAssignStorageToManagedTensors()
1380 FastMap<std::string, at::Tensor> managed_tensor_name_to_tensor{ in TEST() local
1387 src, std::move(managed_tensor_name_to_tensor), min_reused_tensors); in TEST()
1403 FastMap<std::string, at::Tensor> managed_tensor_name_to_tensor{ in TEST() local
1411 src, std::move(managed_tensor_name_to_tensor), min_reused_tensors); in TEST()
1430 FastMap<std::string, at::Tensor> managed_tensor_name_to_tensor{ in TEST() local
1434 src, std::move(managed_tensor_name_to_tensor), min_reused_tensors); in TEST()
|