Home
last modified time | relevance | path

Searched refs:host_device (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_head_tail_outside_compilation.cc88 llvm::StringRef host_device) { in CreateLaunchForBlock() argument
109 op->getLoc(), builder->getStringAttr(host_device), launch_result_types); in CreateLaunchForBlock()
183 llvm::StringRef host_device) { in CreateHeadComputation() argument
191 builder, cluster, /*before=*/true, launch_block, host_device); in CreateHeadComputation()
204 std::string* host_device, bool* cluster_updated) { in LiftHeadOutsideCompiledOps() argument
209 host_device))) in LiftHeadOutsideCompiledOps()
213 *host_device); in LiftHeadOutsideCompiledOps()
291 llvm::StringRef host_device) { in CreateTailComputation() argument
299 builder, cluster, /*before=*/false, launch_block, host_device); in CreateTailComputation()
349 const mlir::TF::RuntimeDevices& devices, std::string host_device, in LiftTailOutsideCompiledOps() argument
[all …]
Doutside_compiled_to_host_launch.cc61 void WrapOpInLaunch(Operation* host_op, llvm::StringRef host_device) { in WrapOpInLaunch() argument
65 host_op->getLoc(), builder.getStringAttr(host_device), in WrapOpInLaunch()
89 std::string host_device; in runOnOperation() local
91 &host_device); in runOnOperation()
94 WrapOpInLaunch(op, host_device); in runOnOperation()
Dtpu_extract_outside_compilation.cc183 OpBuilder& builder, Operation* loc_op, llvm::StringRef host_device) { in CreateLaunchOpForOutsideCluster() argument
187 loc_op->getLoc(), builder.getStringAttr(host_device), in CreateLaunchOpForOutsideCluster()
424 llvm::StringRef host_device) { in CreateParallelExecuteForOutsideCompilation() argument
437 CreateLaunchOpForOutsideCluster(builder, tpu_cluster, host_device); in CreateParallelExecuteForOutsideCompilation()
499 std::string host_device; in runOnOperation() local
501 &host_device); in runOnOperation()
503 host_device); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util_test.cc638 std::string host_device; in TEST() local
640 GetHostDeviceOutsideComputation(devices, cluster, &host_device))); in TEST()
659 std::string host_device; in TEST() local
661 GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); in TEST()
679 std::string host_device; in TEST() local
681 GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); in TEST()
699 std::string host_device; in TEST() local
701 GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); in TEST()
722 std::string host_device; in TEST() local
724 GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); in TEST()
[all …]
Dtpu_rewrite_device_util.h248 std::string* host_device);
Dtpu_rewrite_device_util.cc491 std::string* host_device) { in GetHostDeviceOutsideComputation() argument
494 *host_device = tensorflow::kTPUReplicatedHost; in GetHostDeviceOutsideComputation()
540 *host_device = tpu_device_assignment.tpu_devices[0][0].host; in GetHostDeviceOutsideComputation()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc299 const auto host_device = host_devices[i]; in Run() local
301 AddHostDisconnectNode(host_device->name(), input_dependencies, in Run()
307 for (const auto host_device : host_devices) { in Run() local
310 host_device->name(), configuration_node, graph, in Run()
325 for (const auto host_device : host_devices) { in Run() local
327 TF_RETURN_IF_ERROR(AddGlobalTPUArrayNode(host_device->name(), in Run()
385 for (const auto host_device : host_devices) { in Run() local
387 AddHostDisconnectNode(host_device->name(), input_dependencies, in Run()
Ddistributed_tpu_rewrite_helpers.cc124 for (const auto host_device : *host_system_devices) { in GetHostSystemDevices()
125 const auto& parsed_name = host_device->parsed_name(); in GetHostSystemDevices()
/external/crosvm/devices/src/usb/host_backend/
Dmod.rs7 pub mod host_device; module
Dhost_backend_device_provider.rs9 use super::host_device::HostDevice;
192 let host_device = Box::new(HostDevice::new( in handle_attach_device() localVariable
197 let port = self.usb_hub.connect_backend(host_device); in handle_attach_device()
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py845 host_device = device_util.get_host_for_device(tpu_device)
846 self._device_input_worker_devices.setdefault(host_device, [])
847 self._device_input_worker_devices[host_device].append(tpu_device)
848 self._host_input_worker_devices.setdefault(host_device, [])
849 self._host_input_worker_devices[host_device].append(host_device)
1340 return len(set([self._device_assignment.host_device(r)
Dcollective_all_reduce_strategy.py524 host_device = device_util.get_host_for_device(self._worker_device)
526 return input_lib.InputWorkers([(host_device, self.worker_devices)])
529 host_device,
Dmirrored_strategy.py458 (host_device, (host_device,) * len(compute_devices))
459 for host_device, compute_devices in self._input_workers_devices
Dinput_lib_test.py453 host_device = device_util.get_host_for_device(tpu_device)
454 worker_device_pairs.setdefault(host_device, [])
455 worker_device_pairs[host_device].append(tpu_device)
Dtpu_strategy_test.py1215 self.assertEqual("/task:0/device:CPU:0", device_assignment.host_device())
1241 self.assertEqual("/task:0/device:CPU:0", device_assignment.host_device())
Dinput_lib.py1764 host_device = device_util.get_host_for_device(self._worker)
1767 self._dataset, self._devices, source_device=host_device)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt34 name: "host_device"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt34 name: "host_device"
/external/tensorflow/tensorflow/python/training/saving/
Dfunctional_saver.py170 host_device = saveable_object_util.set_cpu0(saveable.device)
171 saveables_by_device.setdefault(host_device, []).append(saveable)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_feed.py707 return device_assignment.host_device(replica=index)
882 device = self._device_assignment.host_device(
Ddevice_assignment.py162 def host_device(self, member in DeviceAssignment
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h375 if (node_.canonical_device().empty() || node_.host_device().empty()) { in trackable()
687 string host_device() const { return node_.host_device(); } in host_device() function
/external/tensorflow/tensorflow/core/profiler/
Dtfprof_log.proto87 string host_device = 7; field