Searched refs:device_capsule (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | pywrap_parallel_device.cc | 52 tensorflow::Safe_PyObjectPtr device_capsule( in PYBIND11_MODULE() local
68 PyTuple_Pack(2, device_capsule.get(), device_info_capsule.get())); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | custom_device_testutil.cc | 62 tensorflow::Safe_PyObjectPtr device_capsule( in PYBIND11_MODULE() local
67 PyTuple_Pack(4, device_capsule.get(), device_info_capsule.get(), in PYBIND11_MODULE()
|
| D | context.py | 1313 def register_custom_device(self, device_capsule, device_name, argument
1317 pywrap_tfe.TFE_Py_RegisterCustomDevice(self._handle, device_capsule,
2703 def register_custom_device(device_capsule, device_name, device_info_capsule): argument
2725 context().register_custom_device(device_capsule, device_name,
|
| /external/tensorflow/tensorflow/python/ |
| D | pywrap_dtensor_device.cc | 87 std::unique_ptr<PyObject, decltype(&PyXDecref)> device_capsule( in PYBIND11_MODULE() local
102 PyTuple_Pack(2, device_capsule.get(), device_info_capsule.get())); in PYBIND11_MODULE()
|