Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for parallel_device (0.16 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
Back to top