Home
last modified time | relevance | path

Searched refs:torch_functions (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_torch_functions_manual.cpp489 void gatherTorchFunctions_0(std::vector<PyMethodDef>& torch_functions);
490 void gatherTorchFunctions_1(std::vector<PyMethodDef>& torch_functions);
491 void gatherTorchFunctions_2(std::vector<PyMethodDef>& torch_functions);
493 void gatherTorchFunctions(std::vector<PyMethodDef>& torch_functions) { in gatherTorchFunctions() argument
496 torch_functions.assign( in gatherTorchFunctions()
500 gatherTorchFunctions_0(torch_functions); in gatherTorchFunctions()
501 gatherTorchFunctions_1(torch_functions); in gatherTorchFunctions()
502 gatherTorchFunctions_2(torch_functions); in gatherTorchFunctions()
513 torch_functions.begin(), in gatherTorchFunctions()
514 torch_functions.end(), in gatherTorchFunctions()
[all …]
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_torch_functions.cpp81 void gatherTorchFunctions${shard_id}(std::vector<PyMethodDef> &torch_functions) {
83 torch_functions.insert(
84 torch_functions.end(),