Home
last modified time | relevance | path

Searched refs:alloc_tensors_by_stealing_from_handles (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dtensor_converter.h22 TORCH_API std::vector<at::Tensor> alloc_tensors_by_stealing_from_handles(
H A Dtensor_converter.cpp17 std::vector<at::Tensor> alloc_tensors_by_stealing_from_handles( in alloc_tensors_by_stealing_from_handles() function
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runner/
H A Dpybind.cpp63 return torch::aot_inductor::alloc_tensors_by_stealing_from_handles( in initAOTIRunnerBindings()
H A Dmodel_container_runner.cpp120 return torch::aot_inductor::alloc_tensors_by_stealing_from_handles( in run()