Home
last modified time | relevance | path

Searched defs:device_capsule (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/parallel_device/
H A Dpywrap_parallel_device.cc52 tensorflow::Safe_PyObjectPtr device_capsule( in PYBIND11_MODULE() local
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcustom_device_testutil.cc62 tensorflow::Safe_PyObjectPtr device_capsule( in PYBIND11_MODULE() local
H A Dcontext.py1313 def register_custom_device(self, device_capsule, device_name, argument
2703 def register_custom_device(device_capsule, device_name, device_info_capsule): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_dtensor_device.cc87 std::unique_ptr<PyObject, decltype(&PyXDecref)> device_capsule( in PYBIND11_MODULE() local