| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 66 def copy_to_device(target_device, source_device="/cpu:0"): argument
91 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
|
| /external/pytorch/aten/src/ATen/test/ |
| D | xpu_device_test.cpp | 43 int target_device = 0; in TEST() local
|
| /external/pytorch/aten/src/ATen/ops/ |
| D | from_blob.h | 53 TensorMaker& target_device(std::optional<Device> value) noexcept { in target_device() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 461 FunctionData(const string& target_device, in FunctionData()
468 const string& target_device() { return target_device_; } in target_device() function
|
| D | process_function_library_runtime.cc | 121 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
|
| D | function.cc | 730 Device* target_device; in IsLocalTarget() local
1036 string target_device = parent_->GetDeviceName(handle); in RunRemote() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device_op_handler.cc | 71 auto target_device = op->DeviceName(); in Execute() local
|
| /external/pytorch/torch/nn/parallel/ |
| D | _functions.py | 58 def forward(ctx, target_device, dim, *inputs): argument
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.h | 280 Status ShouldUseRemoteExecutionForFn(const std::string& target_device, in ShouldUseRemoteExecutionForFn()
|
| D | tpu_functional_ops.cc | 1597 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
|
| /external/pytorch/torch/distributed/ |
| D | utils.py | 111 def _recursive_to(inputs, target_device, use_side_stream_for_tensor_copies): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 303 const string& target_device = function_target.first; in ComputeAsync() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.h | 79 string target_device; member
|
| D | generic_layout_optimizer_transposer.cc | 249 absl::string_view target_device, absl::string_view src_format, in AssignDeviceAndDataFormats()
|
| /external/pytorch/torch/csrc/utils/ |
| D | tensor_numpy.cpp | 476 const auto target_device = [&]() -> std::optional<Device> { in tensor_from_cuda_array_interface() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 320 const Device* target_device, EagerContext* ctx, in SerializePackedHandle()
|
| /external/mesa3d/src/egl/drivers/dri2/ |
| D | egl_dri2.h | 209 dev_t target_device; member
|
| /external/pytorch/benchmarks/dynamo/ |
| D | common.py | 1309 def deepcopy_model_and_inputs_to_device(self, model, example_inputs, target_device): argument
|
| /external/mesa3d/src/vulkan/wsi/ |
| D | wsi_common_wayland.c | 88 dev_t target_device; member
|