Searched refs:tensor_module (Results 1 – 1 of 1) sorted by relevance
92 auto tensor_module = THPObjectPtr(PyImport_ImportModule("torch._tensor")); in THPAutograd_initExtension() local93 if (!tensor_module) in THPAutograd_initExtension()97 THPVariableClass = PyObject_GetAttrString(tensor_module, "Tensor"); in THPAutograd_initExtension()