Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_head_tail_outside_compilation.cc90 llvm::StringRef host_device) { in CreateLaunchForBlock() argument
111 op->getLoc(), builder->getStringAttr(host_device), launch_result_types); in CreateLaunchForBlock()
188 llvm::StringRef host_device) { in CreateHeadComputation() argument
196 builder, cluster, /*before=*/true, launch_block, host_device); in CreateHeadComputation()
209 std::string* host_device, bool* cluster_updated) { in LiftHeadOutsideCompiledOps() argument
214 host_device))) in LiftHeadOutsideCompiledOps()
218 *host_device); in LiftHeadOutsideCompiledOps()
300 llvm::StringRef host_device) { in CreateTailComputation() argument
308 builder, cluster, /*before=*/false, launch_block, host_device); in CreateTailComputation()
358 const mlir::TF::RuntimeDevices& devices, std::string host_device, in LiftTailOutsideCompiledOps() argument
[all …]
Doutside_compiled_to_host_launch.cc39 void WrapOpInLaunch(Operation* host_op, llvm::StringRef host_device) { in WrapOpInLaunch() argument
43 host_op->getLoc(), builder.getStringAttr(host_device), in WrapOpInLaunch()
63 std::optional<std::string> host_device) { in runOnOperation() argument
66 if (!host_device) { in runOnOperation()
71 WrapOpInLaunch(op, *host_device); in runOnOperation()
Dhost_launch_to_outside_compiled.cc70 std::optional<std::string> host_device) { in runOnOperation() argument
74 if (host_device && device_attr && in runOnOperation()
75 device_attr.getValue().equals(*host_device)) in runOnOperation()
Dtpu_extract_outside_compilation.cc255 OpBuilder& builder, Operation* loc_op, llvm::StringRef host_device) { in CreateLaunchOpForOutsideCluster() argument
259 loc_op->getLoc(), builder.getStringAttr(host_device), in CreateLaunchOpForOutsideCluster()
723 llvm::StringRef host_device) { in CreateParallelExecuteForOutsideCompilation() argument
736 CreateLaunchOpForOutsideCluster(builder, tpu_cluster, host_device); in CreateParallelExecuteForOutsideCompilation()
802 std::string host_device; in runOnOperation() local
804 devices, tpu_cluster, &host_device))) in runOnOperation()
807 host_device))) in runOnOperation()
/external/tensorflow/tensorflow/core/tfrt/utils/
Dgraph_partition.cc74 const std::vector<std::string>& outputs, const Device* host_device, in PrepareSubgraphForFunctionConversion() argument
184 .AssignedDevice(host_device->name()) in PrepareSubgraphForFunctionConversion()
214 const std::string& func_name, const Device* host_device, in BuildPartitionedCallOp() argument
225 call_builder.AssignedDevice(host_device->name()); in BuildPartitionedCallOp()
296 const std::string& stateful_call_func_name, const Device* host_device, in BuildStatefulPartitionedCallOp() argument
302 call_builder.AssignedDevice(host_device->name()); in BuildStatefulPartitionedCallOp()
351 .AssignedDevice(host_device->name()) in BuildStatefulPartitionedCallOp()
421 const Device* host_device, const std::vector<std::string>& inputs, in InsertTransferOps() argument
470 inputs, outputs, host_device, func_name, input_nodes, output_nodes, in InsertTransferOps()
477 BuildPartitionedCallOp(func_name, host_device, device, input_nodes, in InsertTransferOps()
[all …]
Dgraph_partition.h58 const Device* host_device, const std::vector<std::string>& inputs,
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc285 const auto host_device = host_devices[i]; in Run() local
287 AddHostDisconnectNode(host_device->name(), input_dependencies, in Run()
293 for (const auto host_device : host_devices) { in Run() local
296 host_device->name(), configuration_node, graph, in Run()
312 for (const auto host_device : host_devices) { in Run() local
314 TF_RETURN_IF_ERROR(AddGlobalTPUArrayNode(host_device->name(), in Run()
372 for (const auto host_device : host_devices) { in Run() local
374 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/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util_test.cc696 std::string host_device; in TEST() local
698 GetHostDeviceOutsideComputation(devices, cluster, &host_device))); in TEST()
714 std::string host_device; in TEST() local
716 GetHostDeviceOutsideComputation(devices, cluster, &host_device))); in TEST()
734 std::string host_device; in TEST() local
736 GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); in TEST()
754 std::string host_device; in TEST() local
756 GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); in TEST()
777 std::string host_device; in TEST() local
779 GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); in TEST()
[all …]
Dtpu_cluster_util.cc51 std::optional<std::string> host_device; in WalkReachableFromTpuCluster() local
57 host_device = host_device_value; in WalkReachableFromTpuCluster()
59 cluster_to_host[tpu_cluster.getOperation()] = host_device; in WalkReachableFromTpuCluster()
108 std::optional<std::string> host_device) { in WalkReachableFromTpuCluster() argument
Dtpu_rewrite_device_util.h254 std::string* host_device);
Dtpu_rewrite_device_util.cc495 std::string* host_device) { in GetHostDeviceOutsideComputation() argument
498 *host_device = tensorflow::kTPUReplicatedHost; in GetHostDeviceOutsideComputation()
542 *host_device = tpu_device_assignment.tpu_devices[0][0].host; in GetHostDeviceOutsideComputation()
/external/crosvm/devices/src/usb/host_backend/
Dhost_backend_device_provider.rs25 use super::host_device::HostDevice;
199 let host_device = match HostDevice::new( in handle_attach_device() localVariable
204 Ok(host_device) => Box::new(host_device), in handle_attach_device()
211 let port = self.usb_hub.connect_backend(host_device); in handle_attach_device()
Dhost_device.rs80 let mut host_device = HostDevice { in new() localVariable
93 let cur_config = host_device in new()
98 let config_descriptor = host_device in new()
103 host_device.claim_interfaces(&config_descriptor); in new()
105 Ok(host_device) in new()
Dmod.rs7 pub mod host_device; module
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py902 host_device = device_util.get_host_for_device(tpu_device)
903 self._device_input_worker_devices.setdefault(host_device, [])
904 self._device_input_worker_devices[host_device].append(tpu_device)
905 self._host_input_worker_devices.setdefault(host_device, [])
906 self._host_input_worker_devices[host_device].append(host_device)
1279 for host_device in self._device_input_worker_devices.keys():
1280 with ops.device(host_device):
1281 host_to_table[host_device] = next_creator(*args, **kwargs)
1465 return len(set([self._device_assignment.host_device(r)
Dcollective_all_reduce_strategy.py580 host_device = device_util.get_host_for_device(self._worker_device)
582 return input_lib.InputWorkers([(host_device, self.worker_devices)])
585 host_device,
Dmirrored_strategy.py463 (host_device, (host_device,) * len(compute_devices))
464 for host_device, compute_devices in self._input_workers_devices
Dinput_lib.py1672 host_device = device_util.get_host_for_device(device_scope)
1675 worker = host_device
1771 host_device = device_util.get_host_for_device(device_scope)
1777 source_device=host_device,
1784 self._dataset, self._devices, source_device=host_device)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt34 name: "host_device"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.tpu.experimental.-device-assignment.pbtxt34 name: "host_device"
/external/tensorflow/tensorflow/python/checkpoint/
Dfunctional_saver.py233 host_device = saveable_object_util.set_cpu0(tensor.device)
235 .setdefault(host_device, {})
/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/python/tpu/
Dtpu_feed.py707 return device_assignment.host_device(replica=index)
883 device = self._device_assignment.host_device(
/external/tensorflow/tensorflow/core/profiler/
Dtfprof_log.proto87 string host_device = 7; field

12