Searched refs:python_tensor_class (Results 1 – 2 of 2) sorted by relevance
32 PyObject* python_tensor_class);
236 PyObject* python_tensor_class) { in registerPythonTensorClass() argument246 device_to_py_class_[static_cast<size_t>(dev.type())] = python_tensor_class; in registerPythonTensorClass()