Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.h32 PyObject* python_tensor_class);
H A Dpython_variable.cpp236 PyObject* python_tensor_class) { in registerPythonTensorClass() argument
246 device_to_py_class_[static_cast<size_t>(dev.type())] = python_tensor_class; in registerPythonTensorClass()