- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for CopyToParallelDevice (0.28 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.cc
// about things that are taken as inputs on the host or on their // existing device (for multi-device functions). std::unique_ptr<ParallelTensor> parallel_tensor( parallel_device.CopyToParallelDevice( context, absl::get<TFE_TensorHandle*>(input), status)); if (TF_GetCode(status) != TF_OK) return absl::nullopt; parallel_inputs.push_back(parallel_tensor.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
// of the ParallelDevice. // // Sets a bad status and returns a nullptr if `tensor` is already on the // ParallelDevice, or if the individual copies fail. std::unique_ptr<ParallelTensor> CopyToParallelDevice(TFE_Context* context, TFE_TensorHandle* tensor, TF_Status* status) const;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
} } // Necessary for a unique_ptr to a forward-declared type. ParallelDevice::~ParallelDevice() = default; std::unique_ptr<ParallelTensor> ParallelDevice::CopyToParallelDevice( TFE_Context* context, TFE_TensorHandle* tensor, TF_Status* status) const { std::vector<TensorHandlePtr> components; components.reserve(underlying_devices_.size());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1)