Searched refs:named_device (Results 1 – 4 of 4) sorted by relevance
239 NamedParallelDevice* named_device = in CopyToParallelDevice() local241 const ParallelDevice& dev = named_device->device(); in CopyToParallelDevice()245 return ParallelTensorToTensorHandle(named_device->name(), context, in CopyToParallelDevice()294 NamedParallelDevice* named_device = in ParallelDeviceExecute() local306 if (named_device->name() == tensor_handle_device) { in ParallelDeviceExecute()318 ExecuteWithSpecialOps(named_device->device(), named_device->name(), in ParallelDeviceExecute()342 named_device->name(), context, in ParallelDeviceExecute()
101 tensorflow::NamedDevice named_device; in PYBIND11_MODULE() local102 if (!named_device.ParseFromString(std::string(s))) { in PYBIND11_MODULE()107 named_devices.push_back(named_device); in PYBIND11_MODULE()111 for (const auto& named_device : named_devices) { in PYBIND11_MODULE() local112 devices[named_device.name()] = named_device.properties(); in PYBIND11_MODULE()
103 named_device = device_properties_pb2.NamedDevice(108 devices=[named_device])113 estimated_perf = grappler_cluster.EstimatePerformance(named_device)
184 named_device = device_properties_pb2.NamedDevice()185 named_device.name = '/GPU:0'186 named_device.properties.type = 'GPU'187 named_device.properties.num_cores = 24188 named_device.properties.frequency = 1000189 named_device.properties.environment['architecture'] = '4'190 cluster = gcluster.Cluster(devices=[named_device])