Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TFE_TensorHandleBackingDeviceName (0.52 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      // Verify that the mirrors are placed on the component devices.
      std::string first_device = TFE_TensorHandleBackingDeviceName(
          second_components[0].get(), status.get());
      ASSERT_EQ(second_underlying_devices[0], first_device);
      std::string second_device = TFE_TensorHandleBackingDeviceName(
          second_components[1].get(), status.get());
      ASSERT_EQ(second_underlying_devices[1], second_device);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.h

    // a copy, returns the destination device of the copy. Note that the returned
    // device name is not always the device holding the tensor handle's memory. If
    // you want the latter, use TFE_TensorHandleBackingDeviceName. This function
    // will block till the operation that produces `h` has completed.
    TF_CAPI_EXPORT extern const char* TFE_TensorHandleDeviceName(
        TFE_TensorHandle* h, TF_Status* status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  3. tensorflow/c/eager/custom_device_test.cc

      auto value_cleaner = tensorflow::gtl::MakeCleanup(
          [var_value]() { TFE_DeleteTensorHandle(var_value); });
      ASSERT_EQ(tensorflow::string(name),
                tensorflow::string(
                    TFE_TensorHandleBackingDeviceName(var_value, status.get())));
      TFE_TensorHandle* var_value_unpacked =
          UnpackTensorHandle(var_value, status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 27 23:39:24 UTC 2020
    - 18.4K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_distributed_test.cc

      const string composite_device_name =
          "/job:localhost/replica:0/task:0/device:COMPOSITE:0";
      EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status),
                composite_device_name);
      EXPECT_EQ(TFE_TensorHandleBackingDeviceName(packed_handle, status),
                composite_device_name);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
    
      // Register and run a function which returns the sum of 3 variables.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api.cc

      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
      }
      return tensorflow::unwrap(h)->DeviceName(&status->status);
    }
    
    const char* TFE_TensorHandleBackingDeviceName(TFE_TensorHandle* h,
                                                  TF_Status* status) {
      if (h == nullptr) {
        status->status = tensorflow::errors::InvalidArgument("Invalid handle");
        return nullptr;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top