Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dpywrap_parallel_device.cc52 tensorflow::Safe_PyObjectPtr device_capsule( in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/python/eager/
Dcustom_device_testutil.cc63 tensorflow::Safe_PyObjectPtr device_capsule( in PYBIND11_MODULE() local
Dcontext.py1170 def register_custom_device(self, device_capsule, device_name, argument
2423 def register_custom_device(device_capsule, device_name, device_info_capsule): argument