Searched refs:setDeviceMap (Results 1 – 2 of 2) sorted by relevance
118 void setDeviceMap(const std::string& workerName, const DeviceMap& deviceMap) { in setDeviceMap() function
567 .def("_set_device_map", &TensorPipeRpcBackendOptions::setDeviceMap); in rpc_init()