Searched refs:device_info_capsule (Results 1 – 3 of 3) sorted by relevance
59 tensorflow::Safe_PyObjectPtr device_info_capsule( in PYBIND11_MODULE() local65 PyCapsule_SetContext(device_info_capsule.get(), in PYBIND11_MODULE()68 PyTuple_Pack(2, device_capsule.get(), device_info_capsule.get())); in PYBIND11_MODULE()
65 tensorflow::Safe_PyObjectPtr device_info_capsule(PyCapsule_New( in PYBIND11_MODULE() local68 PyTuple_Pack(4, device_capsule.get(), device_info_capsule.get(), in PYBIND11_MODULE()
1171 device_info_capsule): argument1175 device_name, device_info_capsule)2423 def register_custom_device(device_capsule, device_name, device_info_capsule): argument2446 device_info_capsule)