Home
last modified time | relevance | path

Searched refs:assignStorageToManagedTensors (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dmemory_planner.cpp71 std::vector<StorageGroup> assignStorageToManagedTensors( in assignStorageToManagedTensors() function
388 managed_tensors_ = assignStorageToManagedTensors( in StandardMemoryPlanner()
H A Dmemory_planner.h93 TORCH_API std::vector<StorageGroup> assignStorageToManagedTensors(
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_module.cc1361 auto groups = assignStorageToManagedTensors( in testAssignStorageToManagedTensors()