- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for copy_tensor_from_device (0.49 sec)
-
tensorflow/c/eager/custom_device_testutil.cc
bool* executed_flag, TF_Status* status) { TFE_CustomDevice custom_device; custom_device.copy_tensor_to_device = &CopyToLoggingDevice; custom_device.copy_tensor_from_device = &CopyTensorFromLoggingDevice; custom_device.delete_device = &DeleteLoggingDevice; custom_device.execute = &LoggingDeviceExecute; LoggingDevice* device = new LoggingDevice;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 03 20:47:31 UTC 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
"tensors for each device into a parallel tensor explicitly.") .c_str()); return nullptr; } // For TFE_CustomDevice::copy_tensor_from_device in the parallel device // registration. // // Currently this is an error, and un-packing ParallelTensors must be performed // explicitly by running a TPUReplicatedOutput operation on the parallel device. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 29 22:05:31 UTC 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
void* device_info); // Method to copy a tensor from the custom device to a target device. TFE_TensorHandle* (*copy_tensor_from_device)(TFE_Context* context, TFE_TensorHandle* tensor, const char* target_device_name,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
const tensorflow::string& target_device_name, ImmediateExecutionTensorHandle** result) override { TF_Status status; handle->Ref(); TFE_TensorHandle* result_handle = device_.copy_tensor_from_device( context_, tensorflow::wrap(handle), target_device_name.c_str(), &status, info_); handle->Unref(); if (!status.status.ok()) return status.status;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0)