Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp476 PyObject* THPModule_inferSize(PyObject* _unused, PyObject* args) { in THPModule_inferSize() function
1327 {"_infer_size", THPModule_inferSize, METH_VARARGS, nullptr},
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1113 def _infer_size(arg1: Size, arg2: Size) -> Size: ... # THPModule_inferSize