Home
last modified time | relevance | path

Searched defs:target_device (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/system/update_engine/payload_consumer/
H A Dpartition_update_generator_android.cc72 std::string target_device; in GenerateOperationsForPartitionsNotInPayload() local
128 const std::string& target_device, in CreatePartitionUpdate()
H A Dfilesystem_verifier_action_unittest.cc448 test_utils::ScopedLoopbackDeviceBinder target_device( in TEST_F() local
509 test_utils::ScopedLoopbackDeviceBinder target_device( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dprefetching_ops.py66 def copy_to_device(target_device, source_device="/cpu:0"): argument
91 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime.h461 FunctionData(const string& target_device, in FunctionData()
468 const string& target_device() { return target_device_; } in target_device() function
H A Dprocess_function_library_runtime.cc121 const string& source_device, const string& target_device, in SendTensors()
140 const string& source_device, const string& target_device, in ReceiveTensorsAsync()
1177 Device* target_device = nullptr; in GetOutputDevices() local
1603 string target_device; in ReleaseHandle() local
1770 string target_device; in RunInternal() local
H A Dfunction.cc730 Device* target_device; in IsLocalTarget() local
1036 string target_device = parent_->GetDeviceName(handle); in RunRemote() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dxpu_device_test.cpp43 int target_device = 0; in TEST() local
/aosp_15_r20/external/pytorch/aten/src/ATen/ops/
H A Dfrom_blob.h53 TensorMaker& target_device(std::optional<Device> value) noexcept { in target_device() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcustom_device_op_handler.cc71 auto target_device = op->DeviceName(); in Execute() local
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A D_functions.py58 def forward(ctx, target_device, dim, *inputs): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.h280 Status ShouldUseRemoteExecutionForFn(const std::string& target_device, in ShouldUseRemoteExecutionForFn()
H A Dtpu_functional_ops.cc1597 const string& target_device = entry.device; in InitializeShardedVarOnTPU() local
2463 const string& target_device, FHandle* handle, in InstantiatePartition()
2719 const string& target_device = entry.device; in ExecuteFunctions() local
/aosp_15_r20/external/pytorch/torch/distributed/
H A Dutils.py111 def _recursive_to(inputs, target_device, use_side_stream_for_tensor_copies): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunction_ops.cc303 const string& target_device = function_target.first; in ComputeAsync() local
/aosp_15_r20/system/core/fs_mgr/libsnapshot/include/libsnapshot/
H A Dsnapshot.h703 std::string target_device; member
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer.h79 string target_device; member
H A Dgeneric_layout_optimizer_transposer.cc249 absl::string_view target_device, absl::string_view src_format, in AssignDeviceAndDataFormats()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_numpy.cpp476 const auto target_device = [&]() -> std::optional<Device> { in tensor_from_cuda_array_interface() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dremote_copy_node.cc320 const Device* target_device, EagerContext* ctx, in SerializePackedHandle()
/aosp_15_r20/system/update_engine/aosp/
H A Ddynamic_partition_control_android.cc903 auto target_device = in PrepareDynamicPartitionsForUpdate() local
/aosp_15_r20/external/mesa3d/src/egl/drivers/dri2/
H A Degl_dri2.h209 dev_t target_device; member
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py1309 def deepcopy_model_and_inputs_to_device(self, model, example_inputs, target_device): argument
/aosp_15_r20/external/mesa3d/src/vulkan/wsi/
H A Dwsi_common_wayland.c86 dev_t target_device; member
/aosp_15_r20/out/soong/.intermediates/build/soong/product_config/
Dproduct_config.json1 ….txt", "BuildNumberFile": "build_number.txt", "BuildHostnameFile": "build_hostname.txt", "BuildThu…