Home
last modified time | relevance | path

Searched defs:managed_tensor_values (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dmemory_planner.cpp28 const c10::FastSet<const Value*>& managed_tensor_values) { in tensorValueToTensor()
208 const auto& managed_tensor_values = block_info.managed_tensor_values(); in MemoryPlanner() local
383 const auto& managed_tensor_values = block_info.managed_tensor_values(); in StandardMemoryPlanner() local
H A Dimpl.h378 const auto& managed_tensor_values() const { in managed_tensor_values() function
H A Dimpl.cpp400 const c10::FastSet<const Value*>& managed_tensor_values) { in ManagedTensorRanges()
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_module.cc1346 FastSet<const Value*> managed_tensor_values; in testAssignStorageToManagedTensors() local