Searched defs:device_info_capsule (Results 1 – 4 of 4) sorted by relevance
59 tensorflow::Safe_PyObjectPtr device_info_capsule( in PYBIND11_MODULE() local
64 tensorflow::Safe_PyObjectPtr device_info_capsule(PyCapsule_New( in PYBIND11_MODULE() local
1314 device_info_capsule): argument2703 def register_custom_device(device_capsule, device_name, device_info_capsule): argument
92 std::unique_ptr<PyObject, decltype(&PyXDecref)> device_info_capsule( in PYBIND11_MODULE() local