Home
last modified time | relevance | path

Searched refs:AllocateParallelDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device.h57 void AllocateParallelDevice(const char* device_name,
Dparallel_device_testlib.h159 tensorflow::parallel_device::AllocateParallelDevice( in RegisterParallelDevice()
Dparallel_device.cc364 void AllocateParallelDevice(const char* device_name, in AllocateParallelDevice() function
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dpywrap_parallel_device.cc55 tensorflow::parallel_device::AllocateParallelDevice( in PYBIND11_MODULE()