Searched refs:device_to_py_class_ (Results 1 – 1 of 1) sorted by relevance
231 static PyObject* device_to_py_class_[static_cast<size_t>( variable241 if (device_to_py_class_[static_cast<size_t>(dev.type())] != nullptr) { in registerPythonTensorClass()246 device_to_py_class_[static_cast<size_t>(dev.type())] = python_tensor_class; in registerPythonTensorClass()250 return device_to_py_class_[static_cast<size_t>(d.type())]; in getPythonTensorClass()