- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for parallel_device (0.16 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.h
#define TENSORFLOW_C_EAGER_PARALLEL_DEVICE_PARALLEL_DEVICE_H_ #include "tensorflow/c/c_api.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" namespace tensorflow { namespace parallel_device { // Allocate a parallel device named `device_name` which forwards operations to // `underlying_devices`, maintaining "parallel tensors" with components placed // on each underlying device. //
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Jun 04 21:49:16 UTC 2020 - 2.9K 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_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) -
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.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_remote_test.cc
#include <array> #include <memory> #include <string> #include "absl/log/log.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/parallel_device/parallel_device_lib.h" #include "tensorflow/c/eager/parallel_device/parallel_device_testlib.h" #include "tensorflow/c/tf_status.h" #include "tensorflow/core/distributed_runtime/master_env.h" #include "tensorflow/core/distributed_runtime/rpc/grpc_server_lib.h"
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Jul 10 07:18:05 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/BUILD
) filegroup( name = "lib_sources", srcs = ["parallel_device_lib.cc"], ) filegroup( name = "device_headers", srcs = ["parallel_device.h"], ) filegroup( name = "device_sources", srcs = ["parallel_device.cc"], ) filegroup( name = "headers", srcs = [ ":device_headers", ":lib_headers", ],
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 5.4K bytes - Viewed (0)