- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for parallel_device_ (0.14 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.cc
std::unique_ptr<ParallelDevice> parallel_device) : device_name_(name), parallel_device_(std::move(parallel_device)) {} const std::string& name() const { return device_name_; } const ParallelDevice& device() const { return *parallel_device_; } private: std::string device_name_; std::unique_ptr<ParallelDevice> parallel_device_; };
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
// ParallelDevice. class ParallelTensor { public: // Construct a ParallelTensor from TensorHandles placed on the component // devices of a ParallelDevice. If called, ParallelTensor::Shape inspects // `components` to determine a shape. static std::unique_ptr<ParallelTensor> FromTensorHandles( const ParallelDevice& parallel_device,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 13.1K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
std::vector<std::string> devices{ "/job:localhost/replica:0/task:0/device:CPU:0", "/job:localhost/replica:0/task:0/device:CPU:1"}; ParallelDevice parallel_device(std::move(devices)); { std::unique_ptr<ParallelTensor> float_tensors = parallel_device.ScalarsFromSequence<float>({10.0, 11.0}, context.get(), status.get());
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
return nullptr; } } return std::unique_ptr<ParallelTensor>( new ParallelTensor(parallel_device, std::move(components), shape, dtype)); } std::unique_ptr<ParallelTensor> ParallelTensor::FromTensorHandles( const ParallelDevice& parallel_device, std::vector<TensorHandlePtr> components, TF_Status* status) { if (components.empty()) { TF_SetStatus(status, TF_INTERNAL,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
#include "tensorflow/c/c_api_experimental.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/immediate_execution_tensor_handle.h" #include "tensorflow/c/eager/parallel_device/parallel_device_lib.h" #include "tensorflow/c/eager/parallel_device/parallel_device_testlib.h" #include "tensorflow/c/eager/tfe_tensorhandle_internal.h" #include "tensorflow/c/tf_buffer.h" #include "tensorflow/c/tf_datatype.h"
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Aug 06 23:56:17 UTC 2024 - 29.4K bytes - Viewed (0)