Home
last modified time | relevance | path

Searched defs:py_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.cpp103 auto py_tensor = py::cast(tensor); in toIValue() local
578 auto py_tensor = py::cast(tensor); in toPyObject() local
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_numpy.cpp193 PyObject* py_tensor = THPVariable_Wrap(prepared_tensor); in tensor_to_numpy() local
H A Dpython_arg_parser.cpp1776 auto py_tensor = py::cast(tensor); in tensor_slow() local