Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for parallel_device_ (0.14 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_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)
  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_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)
  5. 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)
Back to top