Home
last modified time | relevance | path

Searched defs:tensor_value_to_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dmemory_planner.cpp29 c10::FastMap<const Value*, at::Tensor*> tensor_value_to_tensor; in tensorValueToTensor() local
74 const c10::FastMap<const Value*, at::Tensor*>& tensor_value_to_tensor) { in assignStorageToManagedTensors()
385 const auto tensor_value_to_tensor = in StandardMemoryPlanner() local
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_module.cc1289 const FastMap<const Value*, at::Tensor*>& tensor_value_to_tensor, in checkStorageGroups()
1347 FastMap<const Value*, at::Tensor*> tensor_value_to_tensor; in testAssignStorageToManagedTensors() local