Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dpywrap_parallel_device.cc59 tensorflow::Safe_PyObjectPtr device_info_capsule( in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/python/eager/
Dcustom_device_testutil.cc65 tensorflow::Safe_PyObjectPtr device_info_capsule(PyCapsule_New( in PYBIND11_MODULE() local
Dcontext.py1171 device_info_capsule): argument
2423 def register_custom_device(device_capsule, device_name, device_info_capsule): argument