| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | tpu_rewrite_device_util_test.cc | 696 std::string host_device; in TEST() local
714 std::string host_device; in TEST() local
734 std::string host_device; in TEST() local
754 std::string host_device; in TEST() local
777 std::string host_device; in TEST() local
803 std::string host_device; in TEST() local
829 std::string host_device; in TEST() local
858 std::string host_device; in TEST() local
|
| H A D | tpu_cluster_util.cc | 51 std::optional<std::string> host_device; in WalkReachableFromTpuCluster() local
108 std::optional<std::string> host_device) { in WalkReachableFromTpuCluster()
|
| H A D | tpu_rewrite_device_util.cc | 495 std::string* host_device) { in GetHostDeviceOutsideComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_extract_head_tail_outside_compilation.cc | 90 llvm::StringRef host_device) { in CreateLaunchForBlock()
188 llvm::StringRef host_device) { in CreateHeadComputation()
209 std::string* host_device, bool* cluster_updated) { in LiftHeadOutsideCompiledOps()
300 llvm::StringRef host_device) { in CreateTailComputation()
358 const mlir::TF::RuntimeDevices& devices, std::string host_device, in LiftTailOutsideCompiledOps()
476 std::string host_device; in runOnOperation() local
|
| H A D | outside_compiled_to_host_launch.cc | 39 void WrapOpInLaunch(Operation* host_op, llvm::StringRef host_device) { in WrapOpInLaunch()
63 std::optional<std::string> host_device) { in runOnOperation()
|
| H A D | tpu_extract_outside_compilation.cc | 255 OpBuilder& builder, Operation* loc_op, llvm::StringRef host_device) { in CreateLaunchOpForOutsideCluster()
723 llvm::StringRef host_device) { in CreateParallelExecuteForOutsideCompilation()
802 std::string host_device; in runOnOperation() local
|
| H A D | host_launch_to_outside_compiled.cc | 70 std::optional<std::string> host_device) { in runOnOperation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_configuration_rewrite_pass.cc | 285 const auto host_device = host_devices[i]; in Run() local
293 for (const auto host_device : host_devices) { in Run() local
312 for (const auto host_device : host_devices) { in Run() local
372 for (const auto host_device : host_devices) { in Run() local
|
| H A D | distributed_tpu_rewrite_pass.cc | 655 string host_device; member
3107 string host_device; in CreatePartitionedDummyVarArgs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | graph_partition.cc | 74 const std::vector<std::string>& outputs, const Device* host_device, in PrepareSubgraphForFunctionConversion()
214 const std::string& func_name, const Device* host_device, in BuildPartitionedCallOp()
296 const std::string& stateful_call_func_name, const Device* host_device, in BuildStatefulPartitionedCallOp()
421 const Device* host_device, const std::vector<std::string>& inputs, in InsertTransferOps()
|
| /aosp_15_r20/external/crosvm/devices/src/usb/backend/host_backend/ |
| H A D | mod.rs | 6 pub mod host_device; module
|
| H A D | host_backend_device_provider.rs | 26 let host_device = HostDevice::new(Arc::new(Mutex::new(device)), device_state)?; in attach_host_backend_device() localVariable
|
| H A D | host_device.rs | 62 let mut host_device = HostDevice { in new() localVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | device_assignment.py | 157 def host_device(self, member in DeviceAssignment
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/ |
| H A D | tfprof_log.proto | 87 string host_device = 7; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/ |
| H A D | tfprof_node.h | 687 string host_device() const { return node_.host_device(); } in host_device() function
|