Searched refs:tensor_value_to_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | memory_planner.cpp | 29 c10::FastMap<const Value*, at::Tensor*> tensor_value_to_tensor; in tensorValueToTensor() local
47 tensor_value_to_tensor.emplace( in tensorValueToTensor()
54 return tensor_value_to_tensor; in tensorValueToTensor()
74 const c10::FastMap<const Value*, at::Tensor*>& tensor_value_to_tensor) { in assignStorageToManagedTensors() argument
85 auto* tensor_ptr = tensor_value_to_tensor.at(value); in assignStorageToManagedTensors()
94 auto* tensor_ptr = tensor_value_to_tensor.at(value); in assignStorageToManagedTensors()
100 return tensor_value_to_tensor.find(value) != tensor_value_to_tensor.end(); in assignStorageToManagedTensors()
385 const auto tensor_value_to_tensor = in StandardMemoryPlanner() local
391 tensor_value_to_tensor); in StandardMemoryPlanner()
393 for (auto& tensor : tensor_value_to_tensor) { in StandardMemoryPlanner()
|
| H A D | memory_planner.h | 96 const c10::FastMap<const Value*, at::Tensor*>& tensor_value_to_tensor);
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_module.cc | 1289 const FastMap<const Value*, at::Tensor*>& tensor_value_to_tensor, in checkStorageGroups() argument
1296 for (auto& key_value : tensor_value_to_tensor) { in checkStorageGroups()
1347 FastMap<const Value*, at::Tensor*> tensor_value_to_tensor; in testAssignStorageToManagedTensors() local
1354 tensor_value_to_tensor.emplace(vmap_it->second, &key_value.second); in testAssignStorageToManagedTensors()
1356 ASSERT_EQ(managed_tensor_values.size(), tensor_value_to_tensor.size()); in testAssignStorageToManagedTensors()
1362 graph->block()->nodes(), ranges, tensor_value_to_tensor); in testAssignStorageToManagedTensors()
1365 groups, ranges, tensor_value_to_tensor, min_reused_tensors); in testAssignStorageToManagedTensors()
|